Machine Learning with Scikit-Learn
Machine Learning with Scikit-Learn
Introduction
Scikit-Learn is a popular Python library for machine learning, offering tools for data
preprocessing, model selection, and evaluation.
Installing Scikit-Learn
pip install scikit-learn
Data Preparation
data = {'Experience': [1, 2, 3, 4, 5], 'Salary': [30000, 35000, 40000, 45000, 50000]}
df = pd.DataFrame(data)
X = df[['Experience']]
y = df['Salary']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
Feature Scaling
scaler = StandardScaler()
X_train_scaled = scaler.fit_transform(X_train)
X_test_scaled = scaler.transform(X_test)
Model Evaluation
mse = mean_squared_error(y_test, predictions)
print("Mean Squared Error:", mse)
Conclusion