Recommended models

class datarobot.models.ModelRecommendation

A collection of information about a recommended model for a project.

Variables:

• project_id (str) – the id of the project the model belongs to

• model_id (str) – the id of the recommended model

• recommendation_type (str) – the type of model recommendation

classmethod get(project_id, recommendation_type=None)

Retrieves the default or specified by recommendation_type recommendation.

Parameters:

• project_id (str) – The project’s id.

• recommendation_type (enums.RECOMMENDED_MODEL_TYPE) – The type of recommendation to get. If None, returns the default recommendation.

Returns:

recommended_model

Return type:

ModelRecommendation

classmethod get_all(project_id)

Retrieves all of the current recommended models for the project.

Parameters:

project_id (str) – The project’s id.

Returns:

recommended_models

Return type:

list of ModelRecommendation

classmethod get_recommendation(recommended_models, recommendation_type)

Returns the model in the given list with the requested type.

Parameters:

• recommended_models (list of ModelRecommendation)

• recommendation_type (enums.RECOMMENDED_MODEL_TYPE) – the type of model to extract from the recommended_models list

Returns:

recommended_model

Return type:

ModelRecommendation or None if no model with the requested type exists

get_model()

Returns the Model associated with this ModelRecommendation.

Returns:

recommended_model

Return type:

Model or DatetimeModel if the project is datetime-partitioned