使用joblib.dump保存模型文件后,是否可以获取模型中使用的功能/变量列表?

问题描述 投票:-1回答:1

我已经构建了scikit-learn kmeans模型,并使用joblib.dump命令将其转储。现在我想用新的数据集测试它,但是不能回忆起构建中使用的功能。任何人都可以帮助我使用哪个模型属性/函数给出模型构建期间使用的功能/变量列表?

machine-learning scikit-learn k-means joblib
1个回答
1
投票

您需要使用完全相同的变量和预处理步骤来创建要使用的模型的功能。此信息是预处理管道的工件,不存储在模型中。因此,通常不能从拾取的模型中恢复。

使用源代码控制(如git)来跟踪预处理代码和模型设置。

© www.soinside.com 2019 - 2024. All rights reserved.