tceron commited on
Commit
174a37c
·
verified ·
1 Parent(s): 1260092

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -1
README.md CHANGED
@@ -3,9 +3,26 @@ library_name: transformers
3
  tags: []
4
  ---
5
 
 
 
 
 
 
 
 
 
 
 
 
6
  # Model Card for Model ID
7
 
8
- <!-- Provide a quick summary of what the model is/does. -->
 
 
 
 
 
 
9
 
10
 
11
 
 
3
  tags: []
4
  ---
5
 
6
+ # Run the classifier:
7
+
8
+ python classi-infoseek/run_hf_info_seek_classifier.py --input_csv /path/to/new_dataset.csv --output_csv /path/to/preds_new_dataset.csv
9
+
10
+
11
+ Required column: "content" which is the query.
12
+ Optional column: prompt_id (if missing, it auto-creates one from row index)
13
+ Output columns: prompt_id, prediction_id, prediction
14
+
15
+
16
+
17
  # Model Card for Model ID
18
 
19
+ This classifier has been trained with ModernBERT large. It classifies user interactions with LLMs into 5 categories:
20
+
21
+ 0: "information seeking",
22
+ 1: "content creation",
23
+ 2: "coding",
24
+ 3: "not english",
25
+ 4: "no request",
26
 
27
 
28