sponsorblock-ml / src /predict.py

Commit History

Move `seconds_to_time` to shared
bb58e90

Joshua Lochner commited on

Use classifier category if transformer generates unknown category
ad7fc61

Joshua Lochner commited on

Use multiclass classifier to filter predictions
25f1183

Joshua Lochner commited on

Temporarily disable filtering of predictions using classifier
69fe24d

Joshua Lochner commited on

Add functionality to predict self-promo and interaction reminders
90d1f68

Joshua Lochner commited on

Improve prediction output format
74c1216

Joshua Lochner commited on

Ensure `input_ids` are on the correct device when predicting
5dd37ab

Joshua Lochner commited on

Fix prediction and evaluation arguments
9b9ffd0

Joshua Lochner commited on

Add source code
5fbdd3c

Joshua Lochner commited on