HugSib commited on
Commit
60415a5
·
verified ·
1 Parent(s): 36e73a1

Update data/model_handler.py

Browse files
Files changed (1) hide show
  1. 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