Views
No views yet
paraphrase-multilingual-MiniLM-L12-v21from sentence_transformers import SentenceTransformer
2
3# 加载微调后的模型
4model = SentenceTransformer("lyingbarrelhome/setfit-hierarchical-binary-classifier")
5
6# 编码文本
7texts = ["这是一个示例文本", "另一个示例"]
8embeddings = model.encode(texts)1import torch
2import json
3from sklearn.linear_model import LogisticRegression
4
5# 加载模型和分类器
6model = SentenceTransformer("lyingbarrelhome/setfit-hierarchical-binary-classifier")
7
8# 训练逻辑回归分类器（推荐方式）
9# 或者加载预训练的神经网络分类器（如果可用）
10
11# 预测
12def predict_text(text):
13 embedding = model.encode([text])
14 prediction = classifier.predict(embedding)
15 probability = classifier.predict_proba(embedding)
16 return prediction[0], probability[0]@misc{setfit-hierarchical-binary-classifier,
author = {Your Name},
title = {SetFit Hierarchical Binary Classifier},
year = {2024},
url = {https://huggingface.co/lyingbarrelhome/setfit-hierarchical-binary-classifier}
}