Code Documentation: Model Finder¶
mastml.legos.model_finder Module¶
This module provides a name_to_constructor dict for all models/estimators in scikit-learn, plus a couple test models and error handling functions
Functions¶
check_models_mixed(model_names) |
Method used to check whether the user has mixed regression and classification tasks |
find_model(model_name) |
Method used to check model names conform to scikit-learn model/estimator names |
Classes¶
AlwaysFive([constant]) |
Class used as a test model that always predicts a value of 5. |
EnsembleRegressor(n_estimators, num_samples, …) |
|
KerasRegressor(conf_dict) |
|
ModelImport(model_path) |
Class used to import pickled models from previous machine learning fits |
RandomGuesser() |
Class used as a test model that always predicts random values for y data. |
Class Inheritance Diagram¶
