Models Module
databricks.DataBricksTeacherFewShotHTC(...) | Databricks-based teacher model using few-shot prompting for hierarchical text classification. |
databricks.DataBricksTeacherZeroShotCOTHTC(...) | Databricks-based teacher model using zero-shot chain-of-thought reasoning. |
huggingface.HuggingFaceStudentHTC(taxonomy, ...) | Hugging Face-based student model for hierarchical text classification. |
huggingface.HuggingfaceTextEncoder(model_name) | Text encoder using Hugging Face sentence-transformers models. |
huggingface.HuggingFaceTeacherFewShotHTC(...) | Hugging Face-based teacher model using few-shot prompting for hierarchical text classification. |
huggingface.HuggingFaceTeacherZeroShotCOTHTC(...) | The huggingface model teacher hierarchical text classifier with zero-shot COT reasoning. |
openai.OpenAITeacherFewShotHTC(taxonomy, config) | OpenAI-based teacher model using few-shot prompting for hierarchical text classification. |
openai.OpenAITeacherZeroShotCOTHTC(taxonomy, ...) | OpenAI-based teacher model using zero-shot chain-of-thought reasoning. |
openai.establish_openai_connection | |
openai.chat | |
openai.format_messages_for_chat |