mgyigit commited on
Commit
d004947
·
verified ·
1 Parent(s): 3d57876

Update src/saving_utils.py

Browse files
Files changed (1) hide show
  1. src/saving_utils.py +4 -4
src/saving_utils.py CHANGED
@@ -7,7 +7,7 @@ script_dir = os.path.dirname(os.path.abspath(__file__))
7
  sys.path.append('..')
8
  sys.path.append('.')
9
 
10
- def save_similarity_output(output_dict, method_name, leaderboard_path="/home/user/app/src/data/leaderboard_results.csv", similarity_path="./data/similarity_results.csv"):
11
  # Load or initialize the DataFrames
12
  print(script_dir)
13
  if os.path.exists(leaderboard_path):
@@ -68,7 +68,7 @@ def save_similarity_output(output_dict, method_name, leaderboard_path="/home/use
68
 
69
  return 0
70
 
71
- def save_function_output(model_output, method_name, func_results_path="./data/function_results.csv", leaderboard_path="./data/leaderboard_results.csv"):
72
  # Load or initialize the DataFrames
73
  if os.path.exists(func_results_path):
74
  func_results_df = pd.read_csv(func_results_path)
@@ -133,7 +133,7 @@ def save_function_output(model_output, method_name, func_results_path="./data/fu
133
 
134
  return 0
135
 
136
- def save_family_output(model_output, method_name, leaderboard_path="./data/leaderboard_results.csv", family_results_path="./data/family_results.csv"):
137
  # Load or initialize the DataFrames
138
  if os.path.exists(leaderboard_path):
139
  leaderboard_df = pd.read_csv(leaderboard_path)
@@ -170,7 +170,7 @@ def save_family_output(model_output, method_name, leaderboard_path="./data/leade
170
 
171
  return leaderboard_df, family_results_df
172
 
173
- def save_affinity_output(model_output, method_name, leaderboard_path="./data/leaderboard_results.csv", affinity_results_path="./data/affinity_results.csv"):
174
  # Load or initialize DataFrames
175
  if os.path.exists(leaderboard_path):
176
  leaderboard_df = pd.read_csv(leaderboard_path)
 
7
  sys.path.append('..')
8
  sys.path.append('.')
9
 
10
+ def save_similarity_output(output_dict, method_name, leaderboard_path="/home/user/app/src/data/leaderboard_results.csv", similarity_path="/home/user/app/src/data/similarity_results.csv"):
11
  # Load or initialize the DataFrames
12
  print(script_dir)
13
  if os.path.exists(leaderboard_path):
 
68
 
69
  return 0
70
 
71
+ def save_function_output(model_output, method_name, func_results_path="/home/user/app/src/data/function_results.csv", leaderboard_path="/home/user/app/src/data/leaderboard_results.csv"):
72
  # Load or initialize the DataFrames
73
  if os.path.exists(func_results_path):
74
  func_results_df = pd.read_csv(func_results_path)
 
133
 
134
  return 0
135
 
136
+ def save_family_output(model_output, method_name, leaderboard_path="/home/user/app/src/data/leaderboard_results.csv", family_results_path="/home/user/app/src/data/family_results.csv"):
137
  # Load or initialize the DataFrames
138
  if os.path.exists(leaderboard_path):
139
  leaderboard_df = pd.read_csv(leaderboard_path)
 
170
 
171
  return leaderboard_df, family_results_df
172
 
173
+ def save_affinity_output(model_output, method_name, leaderboard_path="/home/user/app/src/data/leaderboard_results.csv", affinity_results_path="/home/user/app/src/data/affinity_results.csv"):
174
  # Load or initialize DataFrames
175
  if os.path.exists(leaderboard_path):
176
  leaderboard_df = pd.read_csv(leaderboard_path)