Spaces:
Running
Running
Update data/model_handler.py
Browse files- data/model_handler.py +2 -3
data/model_handler.py
CHANGED
@@ -55,7 +55,7 @@ class ModelHandler:
|
|
55 |
print(f"Error loading {model_name} - {e}")
|
56 |
continue
|
57 |
|
58 |
-
self._save_model_infos()
|
59 |
|
60 |
model_res = {}
|
61 |
if len(self.model_infos) > 0:
|
@@ -79,6 +79,7 @@ class ModelHandler:
|
|
79 |
|
80 |
@staticmethod
|
81 |
def add_rank(df):
|
|
|
82 |
cols_to_rank = [
|
83 |
col
|
84 |
for col in df.columns
|
@@ -101,6 +102,4 @@ class ModelHandler:
|
|
101 |
for col in df.columns:
|
102 |
if df[col].dtype == "float64":
|
103 |
df[col] = df[col].apply(lambda x: round(x * 100, 1))
|
104 |
-
# Fill NaN after averaging
|
105 |
-
df.fillna("", inplace=True)
|
106 |
return df
|
|
|
55 |
print(f"Error loading {model_name} - {e}")
|
56 |
continue
|
57 |
|
58 |
+
#self._save_model_infos()
|
59 |
|
60 |
model_res = {}
|
61 |
if len(self.model_infos) > 0:
|
|
|
79 |
|
80 |
@staticmethod
|
81 |
def add_rank(df):
|
82 |
+
df.fillna(0.0, inplace=True)
|
83 |
cols_to_rank = [
|
84 |
col
|
85 |
for col in df.columns
|
|
|
102 |
for col in df.columns:
|
103 |
if df[col].dtype == "float64":
|
104 |
df[col] = df[col].apply(lambda x: round(x * 100, 1))
|
|
|
|
|
105 |
return df
|