Skip to content

Commit

Permalink
updated everything
Browse files Browse the repository at this point in the history
  • Loading branch information
Aarize29 committed Jul 6, 2024
1 parent 789a190 commit 164309a
Show file tree
Hide file tree
Showing 5 changed files with 10 additions and 9 deletions.
6 changes: 3 additions & 3 deletions output/Gradient Boosting_classification_report.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@ Gradient Boosting Classification Report:
precision recall f1-score support

0 0.93 0.91 0.92 1152
1 0.78 0.82 0.80 432
1 0.77 0.81 0.79 432

accuracy 0.89 1584
accuracy 0.88 1584
macro avg 0.85 0.86 0.86 1584
weighted avg 0.89 0.89 0.89 1584
weighted avg 0.89 0.88 0.89 1584

Binary file modified output/Gradient Boosting_confusion_matrix.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
10 changes: 5 additions & 5 deletions output/Random Forest_classification_report.txt
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
Random Forest Classification Report:
precision recall f1-score support

0 0.93 0.93 0.93 1152
1 0.81 0.82 0.81 432
0 0.93 0.92 0.92 1152
1 0.79 0.81 0.80 432

accuracy 0.90 1584
macro avg 0.87 0.87 0.87 1584
weighted avg 0.90 0.90 0.90 1584
accuracy 0.89 1584
macro avg 0.86 0.87 0.86 1584
weighted avg 0.89 0.89 0.89 1584

Binary file modified output/Random Forest_confusion_matrix.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
3 changes: 2 additions & 1 deletion sentiment_analysis.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,5 +74,6 @@
best_pipeline.fit(df['tweet'], df['label'])


joblib.dump(best_pipeline, os.path.join(output_dir, 'best_model.pkl'))

joblib.dump(best_pipeline, 'best_model.pkl')
print(f'The best model is {best_model_name} with accuracy {results[best_model_name]}')

0 comments on commit 164309a

Please sign in to comment.