Zeqhx commited on
Commit
e02c0b8
·
verified ·
1 Parent(s): 99c76a8

Retrain from scratch on dataset 4 (notebook pipeline)

Browse files
Files changed (3) hide show
  1. README.md +33 -0
  2. model.safetensors +1 -1
  3. special_tokens_map.json +35 -5
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language: en
4
+ pipeline_tag: token-classification
5
+ tags:
6
+ - ner
7
+ - resume-parsing
8
+ - cv-parser
9
+ base_model: bert-base-uncased
10
+ ---
11
+
12
+ # CV Parser NER — bert-base-uncased (v2)
13
+
14
+ Token-classification model that extracts **Job Titles**, **Skills**, and
15
+ **Education** from resumes/CVs using a BIO tag scheme.
16
+
17
+ ## Provenance
18
+ - **Trained from scratch on dataset 4** (`resume_bio_annotated_full.csv`,
19
+ 2,483 resumes — 1,739 train / 372 val / 372 test), the team's finalized
20
+ AI-Studio/Vertex-relabelled dataset.
21
+ - Reproduced end-to-end with the project notebooks/scripts
22
+ (`retokenize.py` + `train_bert_run.py`).
23
+ - Base model: `bert-base-uncased` · epochs: 4 · learning rate: 2e-5 ·
24
+ max_length 512 · stride 128 · seed 42.
25
+
26
+ ## Resume-level performance (dataset-4 splits)
27
+ | split | precision | recall | F1 |
28
+ |-------|-----------|--------|----|
29
+ | validation | — | — | 0.5540 |
30
+ | test | — | — | 0.5852 |
31
+
32
+ ## Labels
33
+ `O, B-JOB_TITLE, I-JOB_TITLE, B-SKILL, I-SKILL, B-EDUCATION, I-EDUCATION`
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ab409afd0beec2b7ec63c26311d0b586a14389eab73ddeb2e9e2b380c2a29833
3
  size 435611468
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ffe2e4ababc52af7e9c7e4d4496eaac6ec2a6c5964eed4b82cccc2a7eb0d8959
3
  size 435611468
special_tokens_map.json CHANGED
@@ -1,7 +1,37 @@
1
  {
2
- "cls_token": "[CLS]",
3
- "mask_token": "[MASK]",
4
- "pad_token": "[PAD]",
5
- "sep_token": "[SEP]",
6
- "unk_token": "[UNK]"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  }
 
1
  {
2
+ "cls_token": {
3
+ "content": "[CLS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "mask_token": {
10
+ "content": "[MASK]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "[PAD]",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "sep_token": {
24
+ "content": "[SEP]",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "[UNK]",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
  }