improve scripts

lm-sys · Sep 13, 2023 · b18d5fd · b18d5fd
1 parent 86af4a9
commit b18d5fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/fastchat/serve/monitor/dataset_release_scripts/lmsys_chat_1m/compute_stats.py b/fastchat/serve/monitor/dataset_release_scripts/lmsys_chat_1m/compute_stats.py
@@ -67,7 +67,7 @@
     for x in df["tstamp"]
 ]
 chat_dates_counts = pd.value_counts(chat_dates) * scale
-print(f"Mean #chat: {np.mean(chat_dates_counts):.2f}")
+print(f"mean #chat per day: {np.mean(chat_dates_counts):.2f}")
 
 fig = px.bar(x=chat_dates_counts.index, y=chat_dates_counts)
 fig.update_layout(

diff --git a/fastchat/serve/monitor/dataset_release_scripts/lmsys_chat_1m/upload_hf_dataset.py b/fastchat/serve/monitor/dataset_release_scripts/lmsys_chat_1m/upload_hf_dataset.py
@@ -12,5 +12,6 @@
     args = parser.parse_args()
 
     objs = json.load(open(args.in_file))
+    print(f"#convs: {len(objs)}")
     data = Dataset.from_list(objs)
     data.push_to_hub("lmsys/chat-1m", private=True)