Как проверить, является ли модель sklearn классификатором или регрессором

Question

Есть ли простой способ проверить, решает ли экземпляр модели задачу классификации или регрессии в библиотеке scikit-learn?

Answer 1

Использование sklearn.base.is_classifier и / или 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))

Выход:

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

Answer 2

Полагаю, вы спрашиваете об этом, потому что у вас есть сериализованная модель, тип которой вы не знаете. Откройте файл и выполните

mlType = type(variable_name)

, где variable_name является дескриптором вашей десериализованной модели.

вывод, например,

class 'sklearn.linear_model.base.LinearRegression'