Machine Learning with Python
Scikit-learn is a Python library used to build machine learning models like prediction, classification, and clustering.
pip install scikit-learn
from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression
X = [[1], [2], [3], [4]] y = [2, 4, 6, 8]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
model = LinearRegression() model.fit(X_train, y_train)
prediction = model.predict([[5]]) print(prediction)
- Regression (predict numbers) - Classification (predict categories) - Clustering (group data)
- Linear Regression - Logistic Regression - Decision Tree - KNN - SVM