find_model

mastml.legos.model_finder.find_model(model_name)[source]

Method used to check model names conform to scikit-learn model/estimator names

Args:

model_name: (str), the name of a model/estimator

Returns:

(str), the scikit-learn model name or raises InvalidModel error