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"
]
} |