def predict(self,
predict,
from_data=None,
when={},
model_name='mdsb_model',
breakpoint=PHASE_END):
"""
:param predict:
:param when:
:param model_name:
:return:
"""
if not predict:
raise ValueError('Please provide valid predict value.')
transaction_type = TRANSACTION_PREDICT
from_ds = None if from_data is None else getDS(from_data)
predict_columns = [predict] if type(predict) != type([]) else predict
transaction_metadata = TransactionMetadata()
transaction_metadata.model_name = model_name
transaction_metadata.model_predict_columns = predict_columns
transaction_metadata.model_when_conditions = when
transaction_metadata.type = transaction_type
transaction_metadata.storage_file = self.storage_file
transaction_metadata.from_data = from_ds
transaction = self.session.newTransaction(transaction_metadata,
breakpoint)
return transaction.output_data
def predict(self,
when={},
from_data=None,
model_name='mdsb_model',
breakpoint=PHASE_END,
**kargs):
"""
:param predict:
:param when:
:param model_name:
:return:
"""
transaction_type = TRANSACTION_PREDICT
from_ds = None if from_data is None else getDS(from_data)
transaction_metadata = TransactionMetadata()
transaction_metadata.model_name = model_name
# This will become irrelevant as if we have trained a model with a predict we just need to pass when or from_data
# predict_columns = [predict] if type(predict) != type([]) else predict
# transaction_metadata.model_predict_columns = predict_columns
transaction_metadata.model_when_conditions = when
transaction_metadata.type = transaction_type
transaction_metadata.storage_file = self.storage_file
transaction_metadata.from_data = from_ds
transaction = self.session.newTransaction(transaction_metadata,
breakpoint)
return transaction.output_data
def predict(self,
predict,
from_data=None,
when={},
model_name='mdsb_model',
breakpoint=PHASE_END):
"""
:param predict:
:param when:
:param model_name:
:return:
"""
transaction_type = TRANSACTION_PREDICT
predict_columns = [predict] if type(predict) != type([]) else predict
transaction_metadata = TransactionMetadata()
transaction_metadata.model_name = model_name
transaction_metadata.model_predict_columns = predict_columns
transaction_metadata.model_when_conditions = when
transaction_metadata.type = transaction_type
transaction_metadata.storage_file = self.storage_file
transaction_metadata.from_data = from_data
transaction = self.session.newTransaction(transaction_metadata,
breakpoint)
return transaction.output_data
