moosetape commited on
Commit
32aa6bd
·
0 Parent(s):

Initial commits

Browse files
.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/")