YouTip LogoYouTip

Scikit-learn Tutorial - Getting Started

ML with Scikit-learn

from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

X = [[5.1,3.5],[4.9,3.0],[6.2,3.4],[5.9,3.0]]
y = [0, 0, 1, 1]

X_train, X_test, y_train, y_test = train_test_split(X, y)
model = RandomForestClassifier()
model.fit(X_train, y_train)
print(accuracy_score(y_test, model.predict(X_test)))

Summary

  • fit() trains, predict() predicts
  • Consistent API across all algorithms
← Flask Tutorial - Hello WorldPyTorch Tutorial - Getting Sta β†’