Sanjayraju30 commited on
Commit
12a620b
·
verified ·
1 Parent(s): 6227435

Update risk_model.py

Browse files
Files changed (1) hide show
  1. risk_model.py +1 -29
risk_model.py CHANGED
@@ -29,32 +29,4 @@ def load_model():
29
 
30
  model = load_model()
31
 
32
- def predict_risk(temp, duration):
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_