我在使用Google Colab运行TPOT文档页面提供的MNIST数据集分类示例。然而,训练后我找不到由TPOT获取的最佳管道的代码。
你能帮我找出这个.py文件保存的位置吗?
from tpot import TPOTClassifierfrom sklearn.datasets import load_digitsfrom sklearn.model_selection import train_test_splitdigits = load_digits()X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target, train_size=0.75, test_size=0.25)tpot = TPOTClassifier(generations=5, population_size=50, verbosity=2)tpot.fit(X_train, y_train)print(tpot.score(X_test, y_test))tpot.export('tpot_mnist_pipeline.py')
回答:
它会在你的工作目录中。
我在虚拟机上运行TPOT时也遇到了这个问题。为此,我提交了一些代码,允许你导出到字符串中。
使用:
print(tpot.export())
它会将管道代码打印到屏幕上