Spaces:
Sleeping
Sleeping
Update risk_model.py
Browse files- risk_model.py +1 -29
risk_model.py
CHANGED
@@ -29,32 +29,4 @@ def load_model():
|
|
29 |
|
30 |
model = load_model()
|
31 |
|
32 |
-
def
|
33 |
-
global model
|
34 |
-
pred = model.predict([[temp, duration]])[0]
|
35 |
-
score = max(model.predict_proba([[temp, duration]])[0]) * 100
|
36 |
-
timestamp = get_ist_time()
|
37 |
-
HISTORY.append({
|
38 |
-
"Temperature": temp,
|
39 |
-
"Duration": duration,
|
40 |
-
"Risk": pred,
|
41 |
-
"Confidence": round(score, 2),
|
42 |
-
"Timestamp": timestamp
|
43 |
-
})
|
44 |
-
return pred, round(score, 2), timestamp
|
45 |
-
|
46 |
-
def retrain_model():
|
47 |
-
try:
|
48 |
-
data = pd.read_csv(DATA_PATH)
|
49 |
-
X = data[["temperature", "duration"]]
|
50 |
-
y = data["risk_level"]
|
51 |
-
clf = RandomForestClassifier().fit(X, y)
|
52 |
-
dump(clf, MODEL_PATH)
|
53 |
-
global model
|
54 |
-
model = clf
|
55 |
-
return "✅ Model retrained successfully"
|
56 |
-
except Exception as e:
|
57 |
-
return f"❌ Error: {str(e)}"
|
58 |
-
|
59 |
-
def get_history_df():
|
60 |
-
return pd.DataFrame(HISTORY)
|
|
|
29 |
|
30 |
model = load_model()
|
31 |
|
32 |
+
def predict_risk_
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|