拟合scikit学习决策树和随机森林分类器时出现MemoryError

Question

我有一个具有86k行，5个特征和1个目标列的pandas DataFrame。我正在尝试使用70％的DataFrame作为训练数据来训练DecisionTreeClassifier，并且从fit方法中得到MemoryError。我试过更改一些参数，但我真的不知道是什么导致了错误，所以我不知道如何处理它。 我使用的是Windows 10，内存为8GB。

代码

ASTPathPiece

错误

ASTPath

当我尝试RandomForestClassifier时，总是在进行拟合的行中发生相同的错误。我该怎么解决？

Answer 1

我一直遇到同样的问题。确保您要处理的是分类问题，而不是回归问题。如果目标列是连续的，则可能要使用http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html而不是RandomForestClassifier。