patrick

Spongebob's best friend

Tiny neural net library written from scratch with cupy backend (sorry CPU gang). Still under construction.

from patrick.nn import NN as nn
from patrick.losses import mse_loss
from patrick.activations import leaky_relu
from patrick.layers  import FCLayer as linear

"""
Load and preprocess your data here 
"""

class Model(nn):
    def __init__(self):
        self.layers =  [
                    linear(64,32),  ## input size is 64
                    leaky_relu(),
                    linear(32, 16),
                    leaky_relu(),
                    linear(16,1)  ## output size is 1
                ]

net = Model()

net.fit(
    x_train,    ## your input features, shape: (num_batches, batch_size, input_size)
    y_train,    ## your labels, shape: (num_batches, batch_size, output_size)
    epochs=60,
    learning_rate=0.005, 
    loss = mse_loss
)

"The best way to learn how something works is to make it from scratch"

-Probably Someone