diff --git a/README.md b/README.md index 5e542f4..ed59d6f 100644 --- a/README.md +++ b/README.md @@ -216,12 +216,13 @@ lang_result_ci = summary["mean_per_language/en/f1_macro/ci_margin"] ### Models -| Model Name | Description | Fixed Classifier | +| Model Name | Description | Adaptive Targets | | --- | --- | --- | -| BiEncoderModel | BiEncoder model using sentence-transformers for ranking and classification tasks. | ❌ | -| JobBERTModel | Job-Normalization BiEncoder from Techwolf: https://huggingface.co/TechWolf/JobBERT-v2 | ❌ | -| RndESCOClassificationModel | Random baseline for multi-label classification with random prediction head for ESCO. | ✅ | - +| BiEncoderModel | BiEncoder model using sentence-transformers for ranking and classification tasks. | ✅ | +| JobBERTModel | Job-Normalization BiEncoder from Techwolf: https://huggingface.co/TechWolf/JobBERT-v2 | ✅ | +| ConTeXTMatchModel | ConTeXT-Skill-Extraction-base from Techwolf: https://huggingface.co/TechWolf/ConTeXT-Skill-Extraction-base | ✅ | +| CurriculumMatchModel | CurriculumMatch bi-encoder from Aleksandruz: https://huggingface.co/Aleksandruz/skillmatch-mpnet-curriculum-retriever | ✅ | +| RndESCOClassificationModel | Random baseline for multi-label classification with random prediction head for ESCO. | ❌ | ## Contributing diff --git a/examples/list_available_tasks_and_models.py b/examples/list_available_tasks_and_models.py index 904b7c7..6a2de32 100644 --- a/examples/list_available_tasks_and_models.py +++ b/examples/list_available_tasks_and_models.py @@ -55,7 +55,7 @@ def get_model_properties(model_class: type, registry_name: str, load_full: bool) if load_full: model = ModelRegistry.create(registry_name) props["Model Name"] = model.name - props["Fixed Classifier"] = "✅" if model.classification_label_space is not None else "❌" + props["Adaptive Targets"] = "✅" if model.classification_label_space is None else "❌" props["Description"] = model.description return props @@ -167,7 +167,7 @@ def format_markdown(df: pd.DataFrame) -> str: # Models model_cols = ( - ["Class Name", "Description", "Fixed Classifier"] + ["Class Name", "Description", "Adaptive Targets"] if load_models else ["Class Name", "Description"] ) diff --git a/src/workrb/models/__init__.py b/src/workrb/models/__init__.py index ddd92e8..00cd8ac 100644 --- a/src/workrb/models/__init__.py +++ b/src/workrb/models/__init__.py @@ -5,10 +5,12 @@ from workrb.models.base import ModelInterface from workrb.models.bi_encoder import BiEncoderModel, ConTeXTMatchModel, JobBERTModel from workrb.models.classification_model import RndESCOClassificationModel +from workrb.models.curriculum_encoder import CurriculumMatchModel __all__ = [ "BiEncoderModel", "ConTeXTMatchModel", + "CurriculumMatchModel", "JobBERTModel", "ModelInterface", "RndESCOClassificationModel",