时间戳

首页 > 解决方案 > 如何运行文件夹中的所有模块？

python-3.x - 如何运行文件夹中的所有模块？

问题描述

/machine_learning
dtree.py
lr.py
nb.py
svm.py
/main.py

每个 python 文件都包含一类机器学习方法。在 main.py 中，将 machine_learning 导入为 ml，因此调用每个方法

model = ml.py_name.model_name()

有没有办法让我建立一个包含所有模型类的列表，比如

[ml.svm.svm_ml(), ml.nb.naivebayes(), ml.lr.logisticregression(), ml.dtree.decisiontree()]

我试过了

ml_list = [name for _, name, _ in pkgutil.iter_modules(['machine_learning'])];
print(ml_list);
#["dtree","lr","nb","svm"]

标签： python-3.x

解决方案

  1. 导入您需要的所有模型 -> from sklearn.neighbors import KNeighborsClassifier

  2. 创建列表模型=[]

  3. 将模型添加到列表 -> models.append(KNeighborsClassifier(n_neighbors=3))

  4. 拆分数据以训练测试

  5. 使用 for 循环使您的数据适合模型

对于模型中的模型：model.fit(X, Y)

推荐阅读