TheXeos commited on
Commit
3afd205
1 Parent(s): 0643cfc

Add env data to MM

Browse files
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -27,8 +27,9 @@ class Matchmaking:
27
  :param max_diff: Maximum difference considered between two models' elo
28
  :param matches: Dictionary containing the match history (to later upload as CSV)
29
  """
30
- def __init__(self, models):
31
  self.models = models
 
32
  self.queue = self.models.copy()
33
  self.k = 20
34
  self.max_diff = 500
@@ -95,7 +96,7 @@ class Matchmaking:
95
  df = pd.concat([df, pd.DataFrame([[model.name, model.elo]], columns=['name', 'elo'])])
96
  df.to_csv('elo.csv', index=False)
97
  df_matches = pd.DataFrame(self.matches)
98
- df_matches.to_csv(f"matches/{datetime.now().strftime('%Y-%m-%d_%H-%M-%S_%f')}.csv", index=False)
99
 
100
 
101
  def match(model1, model2) -> float:
 
27
  :param max_diff: Maximum difference considered between two models' elo
28
  :param matches: Dictionary containing the match history (to later upload as CSV)
29
  """
30
+ def __init__(self, models, env):
31
  self.models = models
32
+ self.env = env
33
  self.queue = self.models.copy()
34
  self.k = 20
35
  self.max_diff = 500
 
96
  df = pd.concat([df, pd.DataFrame([[model.name, model.elo]], columns=['name', 'elo'])])
97
  df.to_csv('elo.csv', index=False)
98
  df_matches = pd.DataFrame(self.matches)
99
+ df_matches.to_csv(f"matches/{self.env}__{datetime.now().strftime('%Y-%m-%d_%H-%M-%S_%f')}.csv", index=False)
100
 
101
 
102
  def match(model1, model2) -> float: