fusetools.ml_tools.Train

class fusetools.ml_tools.Train

Bases: object

Functions for training machine learning models.

Methods

train_model

Takes a ScikitLearn estimator instance and train and test dataframes returns: an estimator as well as the predictions and probabilities for the test set.

classmethod train_model(estimator, X_train, X_test, y_train, predict_method)

Takes a ScikitLearn estimator instance and train and test dataframes returns: an estimator as well as the predictions and probabilities for the test set.

Parameters

• estimator – ScikitLearn estimator instance

• X_train – Input training dataset

• X_test – Input test dataset

• y_train – Target training dataset

• predict_method – Type of prediction to make.

Returns

Fitted SckitLearn estimator, test predictions and test scores for each prediction.