feat: house price prediction model integration

This commit is contained in:
nasim
2025-04-03 10:14:54 +02:00
parent 72375db5cd
commit 3aa78be3d6
22 changed files with 251 additions and 24 deletions
+33
View File
@@ -0,0 +1,33 @@
from sklearn.datasets import fetch_california_housing
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
import joblib
import pandas as pd
# Load dataset
data = fetch_california_housing()
df = pd.DataFrame(data.data, columns=data.feature_names)
df['target'] = data.target # in 100k USD
# Engineer features
df['square_feet'] = df['AveRooms'] * 350
df['bedrooms'] = df['AveBedrms']
df['bathrooms'] = df['AveRooms'] * 0.2
# Clean bathrooms
df['bathrooms'] = df['bathrooms'].clip(lower=1)
X = df[['square_feet', 'bedrooms', 'bathrooms']]
y = df['target']
# Train/test split
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=42)
# Train model
model = LinearRegression()
model.fit(X_train, y_train)
# Need to be tested of course..: )
# Save model
joblib.dump(model, 'price_predictor.pkl')
Binary file not shown.