class Neuratron::Model

Direct Known Subclasses

Defined in:

neuratron/model.cr

Instance Method Summary

Instance Method Detail

def compile(loss : Neuratron::Loss | Nil, optimizer : Neuratron::Optimizer | Nil, metrics) #

[View source]
def evaluate #

[View source]
def fit(x_data : Array(LA::GMat), y_data : Array(LA::GMat), epochs : UInt32, batch_size : UInt32, callbacks = Array(Neuratron::Callback).new) #

TODO : Add validation data

TODO : Use callbacks for log


[View source]
def predict(x_data : LA::GMat) #

[View source]