Added train_model.py and made the necessary modifications #19
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
We tested these changes on a private dataset with the following results:
Before:
Predicted Actual | Not Profane(0) | Profane(1)
Not Profane(0) | 703 | 14
Profane(1) | 93 | 39
Accuracy Score: 87.4%
After:
Predicted Actual | Not Profane(0) | Profane(1)
Not Profane(0) | 697 | 20
Profane(1) | 87 | 45
Accuracy Score: 87.4%
Used in production as committed here: https://github.com/dimitrismistriotis/profanity-check