Gumball2k5 commited on
Commit
4c1d219
·
verified ·
1 Parent(s): b31ad32

Update src/benchmark_utils.py

Browse files
Files changed (1) hide show
  1. src/benchmark_utils.py +1 -10
src/benchmark_utils.py CHANGED
@@ -13,22 +13,13 @@ def load_leaderboard(file_path="data/results_df_all_tuned.csv"):
13
  pd.DataFrame: DataFrame đã được sắp xếp, sẵn sàng để hiển thị.
14
  """
15
  try:
16
- # Đọc file CSV
17
  df = pd.read_csv(file_path)
18
-
19
- # --- Xử lý cơ bản (Quan trọng) ---
20
- # Checklist yêu cầu sắp xếp theo RMSE.
21
- # Hãy kiểm tra tên cột RMSE thực tế trong file CSV của bạn.
22
 
23
- # Cập nhật 'TEN_COT_RMSE' thành tên cột chính xác của bạn
24
- # (ví dụ: 'Mean_RMSE', 'Test_RMSE', 'RMSE_CV')
25
- SORT_COLUMN_NAME = 'Mean_RMSE' # <-- THAY ĐỔI TÊN CỘT NÀY NẾU CẦN
26
 
27
  if SORT_COLUMN_NAME in df.columns:
28
- # Sắp xếp các mô hình từ tốt nhất (RMSE thấp nhất) đến tệ nhất
29
  df_sorted = df.sort_values(by=SORT_COLUMN_NAME, ascending=True)
30
  else:
31
- # Nếu không tìm thấy cột, trả về DataFrame gốc và cảnh báo
32
  st.warning(f"Không tìm thấy cột '{SORT_COLUMN_NAME}' để sắp xếp leaderboard. "
33
  f"Vui lòng kiểm tra file `src/benchmark_utils.py`.")
34
  df_sorted = df
 
13
  pd.DataFrame: DataFrame đã được sắp xếp, sẵn sàng để hiển thị.
14
  """
15
  try:
 
16
  df = pd.read_csv(file_path)
 
 
 
 
17
 
18
+ SORT_COLUMN_NAME = 'RMSE (Absolute Error)'
 
 
19
 
20
  if SORT_COLUMN_NAME in df.columns:
 
21
  df_sorted = df.sort_values(by=SORT_COLUMN_NAME, ascending=True)
22
  else:
 
23
  st.warning(f"Không tìm thấy cột '{SORT_COLUMN_NAME}' để sắp xếp leaderboard. "
24
  f"Vui lòng kiểm tra file `src/benchmark_utils.py`.")
25
  df_sorted = df