ONNX 模型库
返回模型

说明文档

基于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