How to check if sklearn model is classifier or regressor

Question

Is there a simple way to check if a model instance solves a classification or regression task in the scikit-learn library?

Answer 1

Use sklearn.base.is_classifier and/or is_regressor:

from sklearn.base import is_classifier, is_regressor
from sklearn.linear_model import LinearRegression
from sklearn.ensemble import RandomForestRegressor
from sklearn.ensemble import RandomForestClassifier

models = [LinearRegression(), RandomForestClassifier(), RandomForestRegressor()]

for m in models:
    print(m.__class__.__name__, is_classifier(m), is_regressor(m))

Output:

# model_name is_classifier is_regressor
LinearRegression False True
RandomForestClassifier True False
RandomForestRegressor False True