时间戳

首页 > 解决方案 > 检查对象是否是库的实例

python - 检查对象是否是库的实例

问题描述

我需要检查一个对象是否来自sklearn库。基本上，我需要检查一个模型是否属于一个特定的库，这样我就可以根据它的类型创建一个通用模式。

我需要检查是否收到了model属于该sklearn库的对象。

例如，

if isinstance(model, sklearn):
    #do something

我想避免尝试深入检查特定模型的类型。

例如，

from sklearn.linear_model import LinearRegression
from sklearn.cluster import FeatureAgglomeration
if isinstance(model, sklearn.linear_model.LinearRegression):
   #to something 

if isinstance(model, sklearn.cluster.FeatureAgglomeration):
   #to something

以上是可接受的型号。但是，sklearn有太多的模型，并且在不断变化。我只想检查它是否来自sklearn图书馆。

标签： pythonscikit-learn

解决方案

没有理想，但你可以使用：

if "sklearn" in str(type(model)):

推荐阅读