kaggle 入门 digit recognizer python randomForestClassifier

kaggl之101新手赛digit recognizer利用python的sklearn包中的randomForestClassifier进行分类

import pandas, numpy
from sklearn.ensemble import RandomForestClassifier

dataset = pandas.read_csv("...\\data\\train.csv")
trainData = dataset.iloc[:, 1:].values
labelData = dataset.label.values
testDataset = pandas.read_csv("...\\data\\test.csv")
testData = testDataset.iloc[:, :].values

rfModel = RandomForestClassifier(n_estimators=100)
rfModel.fit(trainData, labelData)
preds = rfModel.predict(testData)
numpy.savetxt('submission_rf_MultiSoftmax.csv', numpy.c_[range(1, len(testData) + 1), preds], delimiter=',',
              header='ImageId,Label', comments='', fmt='%d')

参考文献：

http://scikit-learn.org/stable/modules/ensemble.html#forest