Skip to content

neeldani/micronet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

micronet

Micronet is a library which gives you the ability to architect any permutations of sequential neural nets. It uses numpy only.

# Example to build your own model

model = Sequential(
    [
        Linear(input_size, hidden_size_1), 
        Relu(),
        Linear(hidden_size_1, hidden_size_2),
        Relu(),
        Linear(hidden_size_2, output_size),
        Softmax(),
    ],
    X = X,
    Y = Y,
    batch_size = 128,
    loss_function = CrossEntropy()
)

model.train(epochs, alpha)

About

Micronet gives you the ability to architect any permutations of sequential neural nets

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages