abacusai.custom_train_function_info
Module Contents
Classes
Information about how to call the customer provided train function. |
- class abacusai.custom_train_function_info.CustomTrainFunctionInfo(client, trainingDataParameterNameMapping=None, schemaMappings=None, trainDataParameterToFeatureGroupIds=None, trainingConfig=None)
Bases:
abacusai.return_class.AbstractApiClass
Information about how to call the customer provided train function.
- Parameters:
client (ApiClient) – An authenticated API Client instance
trainingDataParameterNameMapping (dict) – The mapping from feature group type to the dataframe parameter name
schemaMappings (dict) – The feature type to feature name mapping for each dataframe
trainDataParameterToFeatureGroupIds (dict) – The mapping from the dataframe parameter name to the feature group id backing the data
trainingConfig (dict) – The configs for training
- __repr__()
Return repr(self).