File size: 747 Bytes
745d226 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# 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:
```python
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) |