datto.TrainModel

class datto.TrainModel[source]

Select & train models

__init__()[source]

Methods

__init__()

model_testing(X_train, y_train, model_type, ...)

Gridsearches using a model for best models/params out of a list of commonly used

run_feature_selection(X_train, y_train, k, ...)

Run SelectKBest feature selection for given datasets.

train_in_chunks(X_train, y_train, ...[, ...])

For large datasets, train model in managable chunk sizes

train_test_split_by_ids(df, id_col, ...)

param df