bteodoru commited on
Commit
31ed164
·
verified ·
1 Parent(s): eb99a03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -59
app.py CHANGED
@@ -54,67 +54,15 @@ app = FastAPI()
54
  # )
55
 
56
  # Încărcăm modelul
57
- # try:
58
- # model = joblib.load('rf_model_optim.joblib')
59
- # FEATURE_ORDER = model.feature_names_in_ # Obținem ordinea corectă a caracteristicilor
60
- # print("Model încărcat cu succes! Feature Order:", FEATURE_ORDER)
61
- # except Exception as e:
62
- # print(f"Eroare la încărcarea modelului: {str(e)}")
63
- # model = None # Setăm modelul ca None în caz de eroare
64
- # FEATURE_ORDER = [] # Inițializăm o listă goală pentru a evita erorile ulterioare
65
-
66
  try:
67
- import os
68
- print(f"Director curent: {os.getcwd()}")
69
- print(f"Fișiere în director: {os.listdir('.')}")
70
-
71
- # Verifică dacă fișierul există
72
- model_path = 'rf_model_optim.joblib'
73
-
74
- if not os.path.exists(model_path):
75
- print(f"EROARE: Fișierul {model_path} nu există!")
76
- model = None
77
- FEATURE_ORDER = []
78
- else:
79
- print(f"Fișierul {model_path} există, încercăm să-l încărcăm...")
80
-
81
- # Încarcă modelul cu excepții detaliate
82
- try:
83
- model = joblib.load(model_path)
84
- print(f"Model încărcat! Tip: {type(model)}")
85
-
86
- # Verifică proprietățile modelului
87
- if hasattr(model, 'feature_names_in_'):
88
- FEATURE_ORDER = model.feature_names_in_
89
- print(f"Feature names din model: {FEATURE_ORDER}")
90
- else:
91
- print("Modelul nu are atributul feature_names_in_")
92
- # Setează valorile implicite
93
- FEATURE_ORDER = np.array(['cement_perecent', 'curing_period', 'compaction_rate'])
94
-
95
- # Verifică dacă poate face predicții
96
- test_data = pd.DataFrame({
97
- 'cement_perecent': [5],
98
- 'curing_period': [28],
99
- 'compaction_rate': [1.0]
100
- })
101
- try:
102
- test_pred = model.predict(test_data)
103
- print(f"Test predicție reușit: {test_pred}")
104
- except Exception as pred_e:
105
- print(f"Eroare la predicție de test: {str(pred_e)}")
106
- traceback.print_exc()
107
-
108
- except Exception as load_e:
109
- print(f"Eroare la încărcarea modelului: {str(load_e)}")
110
- traceback.print_exc()
111
- model = None
112
- FEATURE_ORDER = []
113
  except Exception as e:
114
- print(f"Eroare generală: {str(e)}")
115
- traceback.print_exc()
116
- model = None
117
- FEATURE_ORDER = []
118
 
119
 
120
  # # Definim clase personalizate pentru erori
 
54
  # )
55
 
56
  # Încărcăm modelul
 
 
 
 
 
 
 
 
 
57
  try:
58
+ model = joblib.load('rf_model_optim.joblib')
59
+ FEATURE_ORDER = model.feature_names_in_ # Obținem ordinea corectă a caracteristicilor
60
+ print("Model încărcat cu succes! Feature Order:", FEATURE_ORDER)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61
  except Exception as e:
62
+ print(f"Eroare la încărcarea modelului: {str(e)}")
63
+ model = None # Setăm modelul ca None în caz de eroare
64
+ FEATURE_ORDER = [] # Inițializăm o listă goală pentru a evita erorile ulterioare
65
+
66
 
67
 
68
  # # Definim clase personalizate pentru erori