module Torchrb

Constants

VERSION

Public Instance Methods

predict(sample) click to toggle source

# File lib/torchrb.rb, line 26
def predict sample
  Torchrb::Wrapper.for sample.class do |model|
    model.predict sample
  end
end

train(model_class) click to toggle source

# 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