File size: 786 Bytes
88da18c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# Configuration for datasets and models
DATASETS = [
    "bwbayu/job_cv_supervised",
    "cnamuangtoun/resume-job-description-fit",
    "InferencePrince555/Resume-Dataset",
    "jog-description-and-salary-in-indonesia",
    "itjobpostdescriptions",
    "resume-dataset"
]

MODEL_CONFIG = {
    "active_model": "sbert",
    "sbert_path": "trained_models/sbert",
    "glove_path": "trained_models/glove.model",
    "doc2vec_path": "trained_models/doc2vec.model",
    "job_taxonomy": [
        "System Administrator",
        "Database Administrator",
        "Web Developer",
        "Security Analyst",
        "Network Administrator",
        "Data Scientist",
        "DevOps Engineer",
        "Cloud Engineer",
        "Machine Learning Engineer",
        "Software Engineer"
    ]
}