import numpy as np
from sklearn.linear_model import LinearRegression

# Training data: hours studied vs marks scored
X = np.array([[1], [2], [3], [4], [5]])   # hours
y = np.array([35, 45, 55, 65, 75])         # marks

# Create and train the model
model = LinearRegression()
model.fit(X, y)

# Predict for 6 hours of study
prediction = model.predict([[6]])
print(f"Predicted marks for 6 hours: {prediction[0]:.2f}")