# Use multiple inputs for better predictions
data = np.array([[170, 25],  # [height, age]
                 [180, 30],
                 [160, 20],
                 [175, 35]])
weights = np.array([70, 80, 60, 75])

model.fit(data, weights)  # Learn from height AND age
prediction = model.predict([[172, 28]])  # Predict using both features