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)