| # 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) |