Module: Torchrb

Defined in:

lib/torchrb.rb,
lib/torchrb/version.rb

Defined Under Namespace

Modules: NN Classes: DataSet, Lua, ModelBase, Torch, Wrapper

Constant Summary

VERSION =

"0.2.0"

Class Method Summary

• .predict(sample) ⇒ Object
• .train(model_class) ⇒ Object

Class Method Details

.predict(sample) ⇒ Object

# File 'lib/torchrb.rb', line 26

def predict sample
  Torchrb::Wrapper.for sample.class do |model|
    model.predict sample
  end
end

.train(model_class) ⇒ Object

# File 'lib/torchrb.rb', line 19

def train model_class
  Torchrb::Wrapper.for model_class do |model|
    model.load_model_data
    model.train
  end
end