matchcake.ml package¶

Subpackages¶

Submodules¶

matchcake.ml.cross_validation module¶

class matchcake.ml.cross_validation.CrossValidation(estimators: Dict[str, BaseEstimator], x: TensorLike | Tensor | Number, y: TensorLike | Tensor | Number | None = None, *, cv: Any | None = None, cross_validate_kwargs: Dict[str, Any] | None = None)¶

Bases: object

ESTIMATOR_NAME_KEY = 'estimator_name'¶

__init__(estimators: Dict[str, BaseEstimator], x: TensorLike | Tensor | Number, y: TensorLike | Tensor | Number | None = None, *, cv: Any | None = None, cross_validate_kwargs: Dict[str, Any] | None = None)¶

run(verbose=True) → CrossValidationOutput¶

class matchcake.ml.cross_validation.CrossValidationOutput(estimators: Dict[str, BaseEstimator], results_df: DataFrame, *, estimator_name_key: str = 'estimator_name')¶

Bases: object

ESTIMATOR_NAME_KEY = 'estimator_name'¶

__init__(estimators: Dict[str, BaseEstimator], results_df: DataFrame, *, estimator_name_key: str = 'estimator_name')¶

property score_columns: List[str]¶

Module contents¶