def classify(self, model_id, data, production_model=False, batch_size=DEFAULT_BATCH_SIZE, auto_batch=True, retry_if_throttled=True): validate_batch_size(batch_size) url = self.get_detail_url(model_id, action='classify') response = MonkeyLearnResponse() for i in range(0, len(data), batch_size): data_dict = self.remove_none_value({ 'data': data[i:i + batch_size], 'production_model': production_model, }) raw_response = self.make_request('POST', url, data_dict, retry_if_throttled=retry_if_throttled) response.add_raw_response(raw_response) return response
def extract(self, model_id, data, production_model=False, batch_size=DEFAULT_BATCH_SIZE, retry_if_throttled=True, extra_args=None): if extra_args is None: extra_args = {} validate_batch_size(batch_size) url = self.get_detail_url(model_id, action='extract') response = MonkeyLearnResponse() for i in range(0, len(data), batch_size): data_dict = self.remove_none_value({ 'data': data[i:i + batch_size], 'production_model': production_model, }) data_dict.update(extra_args) raw_response = self.make_request('POST', url, data_dict, retry_if_throttled=retry_if_throttled) response.add_raw_response(raw_response) return response