datto.TrainModel
- class datto.TrainModel[source]
Select & train models
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