from sklearn.linear_model import LinearRegression
import pandas as pd
from palmerpenguins import load_penguins
# Load and clean data
penguins = load_penguins()
df = penguins.dropna()
X = df[["bill_length_mm"]]
y = df["body_mass_g"]
# Fit model
model = LinearRegression()
model.fit(X, y)
# Predict
y_pred = model.predict(X)