Module markov.api.recording.experiments.integrations.sklearn.sklearn_auto_record

Functions

def auto_record(name: str, notes: str = '', project_id: str = '', model_class: ModelClass = ModelClass.CLASSIFICATION, **kwargs)

This will record the hyper-parameters of the sklearn estimator automatically when called. Note: This does not record any metrics during the training process currently.

The hyper-parameters are recorded when any of the following methods are called: - fit - fit_transform - fit_predict

Args

name

Name of the experiment being recorded

notes

Notes for the experiment being recorded

project_id

Id of the project where the experiment should be stored

model_class

type of model used for training - Classification, Regression etc. See markov.ModelClass

**kwargs: Returns:

def patched_fit(wrapped, instance, args, kwargs)