Sha256: 205b2cfb306d32359069b5ab00083fbb1463589bdaf4e0cf6a6232458b6232ed

Contents?: true

Size: 1.14 KB

Versions: 1

Compression:

Stored size: 1.14 KB

Contents

require 'big_ml/base'

module BigML
  class Ensemble < Base
    ENSEMBLE_PROPERTIES = [
      :category, :code, :columns, :created, :credits, 
      :dataset, :dataset_status, :description, :dev, :distributions,
      :holdout, :error_models, :excluded_fields, :finished_models,
      :input_fields, :locale, :max_columns, :max_rows, :model_order,
      :models, :name, :node_threshold, :number_of_batchpredictions,
      :number_of_evaluations, :number_of_models, :number_of_predictions,
      :number_of_public_predictions, :objective_field, :ordering,
      :out_of_bag, :price, :private, :randomize, :random_candidates,
      :range, :replacement, :resource, :rows, :sample_rate, :seed,
      :shared, :shared_hash, :sharing_key, :size, :source, :source_status, 
      :status, :subscription, :tags, :tlp, :updated
    ]

    attr_reader *ENSEMBLE_PROPERTIES

    def to_prediction(options)
      Prediction.create(resource, options)
    end

    class << self
      def create(dataset, options = {})
        response = client.post("/#{resource_name}", {}, options.merge(dataset: dataset))
        self.new(response) if response.success?
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
big_ml-0.1.3 lib/big_ml/ensemble.rb