返回模型
说明文档
基于TF-IDF的多项式朴素贝叶斯
这是一个在钓鱼检测数据集上使用TF-IDF向量化器训练的多项式朴素贝叶斯模型。该模型使用scikit-learn训练,并得到了纽卡斯尔大学国家数据创新中心(NICD)的支持。
使用方法
您可以使用以下脚本加载和使用此模型:
import joblib
from huggingface_hub import hf_hub_download
# Download the model from Hugging Face
model_path = hf_hub_download(repo_id="your_username/multi-nb-tfidf-model", filename="multi-nb-tfidf-model.pkl")
# Load the model
model = joblib.load(model_path)
# Use the model for predictions
sample_text = ["Example text to classify"]
predictions = model.predict(sample_text)
print(predictions)
thuggingface/multi-nb-tfidf-model
作者 thuggingface
transformers
↓ 2
♥ 0
创建时间: 2024-07-19 12:50:15+00:00
更新时间: 2024-07-19 15:19:37+00:00
在 Hugging Face 上查看文件 (5)
.gitattributes
README.md
config.json
multi-nb-tfidf-model.onnx
ONNX
multi-nb-tfidf-model.pkl