from sklearn.linear_model import LogisticRegression, LinearRegression, Ridge
from catboost import CatBoostClassifier, CatBoostRegressor


clf_model = 'classif'
regr_model = 'regr'

name2model = {
    'LogisticRegression': (LogisticRegression, clf_model),
    'CatBoostClassifier': (CatBoostClassifier, clf_model),
    'CatBoostRegressor': (CatBoostRegressor, regr_model),
    'LinearRegression': (LinearRegression, regr_model),
    'Ridge': (Ridge, regr_model)
}
