random_forest.py

from mnist import *

from sklearn.ensemble import RandomForestClassifier

forest_clf = RandomForestClassifier(random_state=42)

#y_probas_forest = cross_val_predict(forest_clf, X_train, y_train_5, cv=3, method="predict_proba")

#y_scores_forest = y_probas_forest[:, 1]
# score = proba of positive class
#fpr_forest, tpr_forest, thresholds_forest = roc_curve(y_train_5,y_scores_forest)

#plot_roc_curve(fpr_forest, tpr_forest)
#plt.show()

forest_clf.fit(X_train, y_train)
proba = forest_clf.predict_proba([some_digit])

print(proba)