class Clarification::TrainRequester

Public Instance Methods

add_image_with_concepts(target_url, concept_array) click to toggle source
# File lib/clarification/requester/train_requester.rb, line 4
def add_image_with_concepts(target_url, concept_array)
  uri = uri_builder("inputs")
  body = image_with_concepts_body_builder(target_url, concept_array)
  response = get_response(uri, body)
  return response
end
create_model(name, concept_array) click to toggle source
# File lib/clarification/requester/train_requester.rb, line 11
def create_model(name, concept_array)
  uri = uri_builder("models")
  body = create_model_body_builder(name, concept_array)
  response = get_response(uri, body)
  return response
end
train_model(id) click to toggle source
# File lib/clarification/requester/train_requester.rb, line 18
def train_model(id)
  uri = uri_builder("models/#{id}/versions")
  body = {}
  response = get_response(uri, body)
  return response
end

Private Instance Methods

create_model_body_builder(name, concept_array) click to toggle source
# File lib/clarification/requester/train_requester.rb, line 56
def create_model_body_builder(name, concept_array)
  body_concepts = []

  concept_array.each do |concept|
    body_concepts << {
      id: concept,
    }  
  end

  {
    model: {
      id: name,
      output_info: {
        data: {
          concepts: body_concepts
        },
        output_config: {
          concepts_mutually_exclusive: false,
          closed_environment:false
        }
      }
    }
  }
end
image_with_concepts_body_builder(target_url, concept_array) click to toggle source
# File lib/clarification/requester/train_requester.rb, line 32
def image_with_concepts_body_builder(target_url, concept_array)
  body_concepts = []

  concept_array.each do |concept|
    body_concepts << {
      id: concept,
      value: true
    }  
  end

  {
    inputs: [
      {
        data: {
          image: {
            url: target_url
          },
          concepts: body_concepts
        }
      }
    ]
  }
end
uri_builder(path) click to toggle source
# File lib/clarification/requester/train_requester.rb, line 27
def uri_builder(path)
  url = "#{Clarification::BASE_URL}#{path}"
  URI.parse(url)
end