Downloads · 30 days
5
14% of all-time downloads
multi-nb-tfidf-model
thuggingface/multi-nb-tfidf-model
multi-nb-tfidf-model is a machine learning model from thuggingface. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This is a Multinomial Naive Bayes model trained with a TF-IDF vectorizer on a phishing detection dataset. It was trained using scikit-learn with the support of the National Innovation Centre for Data (NICD) at Newcast…
All-time downloads
36
Public
Repo size
24.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pkl9.7 MB · 67%
At a glance
- Library
- transformers
- Model type
- sklearn
- Access
- Public
- Created
- Jul 19, 2024
- Updated
- Jul 19, 2024
- SHA
- 8c1ac428
Model card
From the Hugging Face model README
Multinomial Naive Bayes with TF-IDF
This is a Multinomial Naive Bayes model trained with a TF-IDF vectorizer on a phishing detection dataset. It was trained using scikit-learn with the support of the National Innovation Centre for Data (NICD) at Newcastle University.
Usage
You can load and use this model with the following script:
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)
- Library
- transformers
- Type
- sklearn
- Created
- Jul 19, 2024
- Updated
- Jul 19, 2024