sklearn random forest mnist code example

Example 1: sklearn random forest

from sklearn.ensemble import RandomForestClassifier


clf = RandomForestClassifier(max_depth=2, random_state=0)

clf.fit(X, y)

print(clf.predict([[0, 0, 0, 0]]))

Example 2: sklearn random forest regressor

from sklearn.ensemble import RandomForestRegressor


clf = RandomForestRegressor(max_depth=2, random_state=0)

clf.fit(X, y)

print(clf.predict([[0, 0, 0, 0]]))