module Monkeylearn::Classifiers

Public Class Methods

build_endpoint(*args) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 19
def build_endpoint(*args)
  File.join('classifiers', *args) + '/'
end
classify(model_id, data, options = {}) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 31
def classify(model_id, data, options = {})
  options[:batch_size] ||= Monkeylearn::Defaults.default_batch_size
  batch_size = options[:batch_size]
  validate_batch_size batch_size

  endpoint = build_endpoint(model_id, 'classify')

  if Monkeylearn.auto_batch
    responses = (0...data.length).step(batch_size).collect do |start_idx|
      sliced_data = { data: data[start_idx, batch_size] }
      if options.key? :production_model
        sliced_data[:production_model] = options[:production_model]
      end
      request(:post, endpoint, sliced_data)
    end

    return Monkeylearn::MultiResponse.new(responses)
  else
    body = {data: data}
    if options.key? :production_model
        body[:production_model] = options[:production_model]
    end
    return request(:post, endpoint, body)
  end
end
create(name, options = {}) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 69
def create(name, options = {})
  data = {
      name: name,
      description: options[:description],
      algorithm: options[:algorithm],
      language: options[:language],
      max_features: options[:max_features],
      ngram_range: options[:ngram_range],
      use_stemming: options[:use_stemming],
      preprocess_numbers: options[:preprocess_numbers],
      preprocess_social_media: options[:preprocess_social_media],
      normalize_weights: options[:normalize_weights],
      stopwords: options[:stopwords],
      whitelist: options[:whitelist],
  }.delete_if { |k,v| v.nil? }
  request(:post, build_endpoint, data)
end
delete(module_id) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 123
def delete(module_id)
  request(:delete, build_endpoint(module_id))
end
deploy(module_id) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 109
def deploy(module_id)
  request(:post, build_endpoint(module_id, 'deploy'))
end
detail(module_id) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 105
def detail(module_id)
  request(:get, build_endpoint(module_id))
end
edit(module_id, options = {}) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 87
def edit(module_id, options = {})
  data = {
      name: options[:name],
      description: options[:description],
      algorithm: options[:algorithm],
      language: options[:language],
      max_features: options[:max_features],
      ngram_range: options[:ngram_range],
      use_stemming: options[:use_stemming],
      preprocess_numbers: options[:preprocess_numbers],
      preprocess_social_media: options[:preprocess_social_media],
      normalize_weights: options[:normalize_weights],
      stopwords: options[:stopwords],
      whitelist: options[:whitelist],
  }.delete_if { |k,v| v.nil? }
  request(:patch, build_endpoint(module_id), data)
end
list(options = {}) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 57
def list(options = {})
  if options.key?(:order_by)
    options[:order_by] = validate_order_by_param(options[:order_by])
  end
  query_params = {
    page: options[:page],
    per_page: options[:per_page],
    order_by: options[:order_by]
  }.delete_if { |k,v| v.nil? }
  request(:get, build_endpoint, nil, query_params)
end
tags() click to toggle source
# File lib/monkeylearn/classifiers.rb, line 15
def tags
  return Tags
end
train(module_id) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 113
def train(module_id)
  request(:post, build_endpoint(module_id, 'train'))
end
upload_data(module_id, data) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 117
def upload_data(module_id, data)
  endpoint = build_endpoint(module_id, 'data')

  request(:post, endpoint, {data: data})
end
validate_batch_size(batch_size) click to toggle source
# File lib/monkeylearn/classifiers.rb, line 23
def validate_batch_size(batch_size)
  max_size = Monkeylearn::Defaults.max_batch_size
  if batch_size >  max_size
    raise MonkeylearnError, "The param batch_size is too big, max value is #{max_size}."
  end
  true
end