abacusai.custom_train_function_info — AbacusAI documentation

# abacusai.custom_train_function_info

## Classes

`AbstractApiClass` |
`CustomTrainFunctionInfo` | Information about how to call the customer provided train function.

## Module Contents

class abacusai.custom_train_function_info.AbstractApiClass(client, id)
  __eq__(other)

    Return self==value.

  _get_attribute_as_dict(attribute)

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).

  to_dict()

    Get a dict representation of the parameters in this class

    Returns:

      The dict value representation of the class parameters

    Return type:

      dict