class Neuratron::Model
- Neuratron::Model
- Reference
- Object
Direct Known Subclasses
Defined in:
neuratron/model.crInstance Method Summary
- #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)
Instance Method Detail
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