abacusai.custom_train_function_info

Classes

CustomTrainFunctionInfo

Information about how to call the customer provided train function.

Module Contents

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

training_data_parameter_name_mapping
schema_mappings
train_data_parameter_to_feature_group_ids
training_config
deprecated_keys
__repr__()
to_dict()

Get a dict representation of the parameters in this class

Returns:

The dict value representation of the class parameters

Return type:

dict