jimnoneill commited on
Commit
c22be37
·
verified ·
1 Parent(s): a731017

Add config.json for HuggingFace download tracking

Browse files
Files changed (1) hide show
  1. config.json +43 -0
config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "pubguard",
3
+ "architecture": "multi_head_linear",
4
+ "embedding_backbone": "minishlab/potion-base-32M",
5
+ "embedding_dim": 512,
6
+ "structural_features": 14,
7
+ "heads": {
8
+ "doc_type": {
9
+ "classes": [
10
+ "scientific_paper",
11
+ "poster",
12
+ "abstract_only",
13
+ "junk"
14
+ ],
15
+ "num_classes": 4,
16
+ "input_dim": 526,
17
+ "accuracy": 0.997,
18
+ "head_file": "models/head_doc_type.npz"
19
+ },
20
+ "ai_detect": {
21
+ "classes": [
22
+ "human",
23
+ "ai_generated"
24
+ ],
25
+ "num_classes": 2,
26
+ "input_dim": 512,
27
+ "accuracy": 0.834,
28
+ "head_file": "models/head_ai_detect.npz"
29
+ },
30
+ "toxicity": {
31
+ "classes": [
32
+ "clean",
33
+ "toxic"
34
+ ],
35
+ "num_classes": 2,
36
+ "input_dim": 512,
37
+ "accuracy": 0.847,
38
+ "head_file": "models/head_toxicity.npz"
39
+ }
40
+ },
41
+ "classifier": "LogisticRegression",
42
+ "library_name": "model2vec"
43
+ }