dansachs commited on
Commit
7f1c111
·
verified ·
1 Parent(s): 9c6b88c

Pushing fine-tuned IndoBERT religious classifier

Browse files
confusion_matrix.png ADDED
label_map.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "Islam": 0,
3
+ "Catholic": 1,
4
+ "Protestant": 2
5
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a83a7287a08dbf7908bc4629ac1b3a942676340ac5c5008611b22dc31b0427ad
3
  size 442265596
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df74510fa71c3d249c81556a17728f747e016fd3bddec530c1849c293c1b4866
3
  size 442265596
training_info.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "indolem/indobert-base-uncased",
3
+ "num_epochs": 3,
4
+ "batch_size": 64,
5
+ "eval_batch_size": 128,
6
+ "max_length": 128,
7
+ "train_samples": 1799998,
8
+ "test_samples": 200000,
9
+ "precision": "bf16",
10
+ "gpu": "NVIDIA A100-SXM4-40GB",
11
+ "label_map": {
12
+ "Islam": 0,
13
+ "Catholic": 1,
14
+ "Protestant": 2
15
+ }
16
+ }