Page 295 - Touhpad Ai

plt.grid(True)

plt.show()

# Perform Linear Regression

# Select features and target
features = ['Year', 'Kilometer', 'Engine', 'Max Power', 'Max Torque']
X = df_cleaned[features]
y = df_cleaned['Price']

# Split data into train and test
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
state=42)

# Train model
model = LinearRegression()
model.fit(X_train, y_train)

# Predict
y_pred = model.predict(X_test)

# Evaluation
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)

# Print Coefficients
coefficients = pd.DataFrame({'Feature': features, 'Coefficient': model.coef_})
print("\nLinear Regression Coefficients:")
print(coefficients)

# Print Evaluation Metrics
print("\nMean Squared Error:",mse)
print("R-squared Score:",r2)

Project Work 293