-
Notifications
You must be signed in to change notification settings - Fork 202
Add spam detection model, training script, and improved Streamlit web… #314
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
… interface with color highlighting for predictions
|
Thank you for submitting your pull request! We'll review it as soon as possible. For further communication, join our discord server https://discord.gg/tSqtvHUJzE. |
Avdhesh-Varshney
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Move your file to models/ChatBot folder. Add need a video which display your code is running in Jarvis. (Means sidebar is verifying it)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove this
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove this
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove all comments & inline css styling. Use only streamlit elements.
For model files like pickle or joblib, you can checkout how movieReccommendationModel is setup.
https://github.com/Code-A2Z/jarvis/blob/main/src/apps/pages/models/Recommendation/movieRecommendationModel.py
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Revert changes from this file
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
So remove this file
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Bro you need to setup your training notebook file on kaggle. And fetch and download the output model file over here as done in movie model
Connect me on discord, if you stuck in somewhere 👍
|
Could you plz raise a cleaned & new PR?
|
|
I'll send a PR today itself, but why did you close the issue? @Avdhesh-Varshney |
|
Major Reason:
Minor
|
… interface with color highlighting for predictions
Closes: #70
Describe the add-ons or changes you've made 📃
#Spam Detection Model Training Script:
Added train_spam_model.py to download, preprocess, balance, and train a spam classifier using the SMS Spam Collection dataset.
The script balances the dataset, trains a Naive Bayes model, evaluates it, and saves both the model and vectorizer as .pkl files.
#Streamlit Web Interface:
Added spam_detection_app.py, a user-friendly web app for spam detection.
Users can input a message and get an instant prediction.
The result is visually highlighted: green for not spam, red for spam, making it easy to interpret.
#Data and Model Files:
The dataset (SMSSpamCollection), trained model (spam_model.pkl), and vectorizer (spam_vectorizer.pkl) are included in the appropriate directory for reproducibility and deployment.
Working Video 📷
Streamlit.-.Google.Chrome.2025-07-29.21-33-16.mp4
###Model Evaluation
Classification Report:
precision recall f1-score support
macro avg 0.98 0.97 0.97 1930
weighted avg 0.98 0.97 0.97 1930
Note to reviewers 📄
Please do check out the changes I have made, and please tell if there is any scope of Improvements