class

Neuratron::Model

Inherits Reference < Object

Instance methods

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

evaluate

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

predict(x_data : LA::GMat)