Spaces:
Sleeping
Sleeping
Commit ·
32aa6bd
0
Parent(s):
Initial commits
Browse files- .gitattributes +2 -0
- README.md +0 -0
- app/_init_.py +0 -0
- app/main.py +18 -0
- data/weather.csv +0 -0
- models/scaler.pkl +0 -0
- models/xgb_model.pkl +0 -0
- scripts/predict.py +19 -0
- scripts/train_model.py +45 -0
.gitattributes
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.png filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
File without changes
|
app/_init_.py
ADDED
|
File without changes
|
app/main.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from fastapi import FastAPI
|
| 2 |
+
import joblib
|
| 3 |
+
import pandas as pd
|
| 4 |
+
|
| 5 |
+
app = FastAPI()
|
| 6 |
+
model = joblib.load('models/xgb_model.pkl')
|
| 7 |
+
scaler = joblib.load('models/scaler.pkl')
|
| 8 |
+
|
| 9 |
+
@app.get("/")
|
| 10 |
+
def home():
|
| 11 |
+
return {"message": "Weather Forecast API"}
|
| 12 |
+
|
| 13 |
+
@app.post("/predict/")
|
| 14 |
+
def predict_weather(data: dict):
|
| 15 |
+
df = pd.DataFrame([data])
|
| 16 |
+
scaled = scaler.transform(df)
|
| 17 |
+
pred = model.predict(scaled)
|
| 18 |
+
return {"predicted_temperature": round(float(pred[0]), 2)}
|
data/weather.csv
ADDED
|
File without changes
|
models/scaler.pkl
ADDED
|
File without changes
|
models/xgb_model.pkl
ADDED
|
File without changes
|
scripts/predict.py
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import joblib
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
model = joblib.load('models/xgb_model.pkl')
|
| 6 |
+
scaler = joblib.load('models/scaler.pkl')
|
| 7 |
+
|
| 8 |
+
# Example: latest 7 days data (replace with real)
|
| 9 |
+
latest_data = pd.DataFrame({
|
| 10 |
+
'humidity': [70, 68, 72, 71, 69, 73, 70],
|
| 11 |
+
'pressure': [1012, 1011, 1013, 1010, 1011, 1014, 1013],
|
| 12 |
+
'windspeed': [8.0, 7.9, 8.1, 8.3, 7.8, 8.2, 8.0],
|
| 13 |
+
'temp_lag1': [24.5], 'temp_lag2': [25.1], 'temp_lag3': [24.8],
|
| 14 |
+
'temp_lag4': [25.0], 'temp_lag5': [24.7], 'temp_lag6': [25.2], 'temp_lag7': [24.9]
|
| 15 |
+
})
|
| 16 |
+
|
| 17 |
+
scaled = scaler.transform(latest_data)
|
| 18 |
+
pred = model.predict(scaled)
|
| 19 |
+
print(f"Predicted temperature: {pred[0]:.2f}°C")
|
scripts/train_model.py
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import numpy as np
|
| 3 |
+
from sklearn.model_selection import train_test_split
|
| 4 |
+
from sklearn.preprocessing import StandardScaler
|
| 5 |
+
from xgboost import XGBRegressor
|
| 6 |
+
from sklearn.metrics import mean_absolute_error, mean_squared_error
|
| 7 |
+
import joblib, os
|
| 8 |
+
|
| 9 |
+
# Load data
|
| 10 |
+
df = pd.read_csv('data/weather.csv')
|
| 11 |
+
df['date'] = pd.to_datetime(df['date'])
|
| 12 |
+
df = df.sort_values('date')
|
| 13 |
+
|
| 14 |
+
# Create lag features (last 7 days)
|
| 15 |
+
for lag in range(1, 8):
|
| 16 |
+
df[f'temp_lag{lag}'] = df['temp'].shift(lag)
|
| 17 |
+
|
| 18 |
+
df = df.dropna()
|
| 19 |
+
|
| 20 |
+
X = df.drop(columns=['date', 'temp'])
|
| 21 |
+
y = df['temp']
|
| 22 |
+
|
| 23 |
+
# Split train/test
|
| 24 |
+
X_train, X_test, y_train, y_test = train_test_split(X, y, shuffle=False, test_size=0.2)
|
| 25 |
+
|
| 26 |
+
# Scale
|
| 27 |
+
scaler = StandardScaler()
|
| 28 |
+
X_train = scaler.fit_transform(X_train)
|
| 29 |
+
X_test = scaler.transform(X_test)
|
| 30 |
+
|
| 31 |
+
# Train model
|
| 32 |
+
model = XGBRegressor(n_estimators=300, learning_rate=0.05, max_depth=6)
|
| 33 |
+
model.fit(X_train, y_train)
|
| 34 |
+
|
| 35 |
+
# Evaluate
|
| 36 |
+
pred = model.predict(X_test)
|
| 37 |
+
mae = mean_absolute_error(y_test, pred)
|
| 38 |
+
rmse = np.sqrt(mean_squared_error(y_test, pred))
|
| 39 |
+
print(f"MAE: {mae:.2f}, RMSE: {rmse:.2f}")
|
| 40 |
+
|
| 41 |
+
# Save model
|
| 42 |
+
os.makedirs('models', exist_ok=True)
|
| 43 |
+
joblib.dump(model, 'models/xgb_model.pkl')
|
| 44 |
+
joblib.dump(scaler, 'models/scaler.pkl')
|
| 45 |
+
print("Model saved to models/")
|