class DGaffML::Model

Attributes

dataset[RW]

Public Class Methods

new(client, model_response) click to toggle source
# File lib/dgaff_ml/model.rb, line 4
def initialize(client, model_response)
  @client = client
  @model = model_response
  @model_id = @model["id"]
  @user_id = @model["user_id"]
end

Public Instance Methods

apply_to_new_dataset(filepath, prediction_column) click to toggle source
# File lib/dgaff_ml/model.rb, line 11
def apply_to_new_dataset(filepath, prediction_column)
  @client.apply_to_new_dataset(@model_id, filepath, prediction_column)
end