sponsorblock-ml / src /shared.py

Commit History

Use logger instead of printing when loading datasets
eaa79a8

Joshua Lochner commited on

Fix training arguments dataclasses
d34e3fe

Joshua Lochner commited on

Remove PreprocessingDatasetArguments class
643d00a

Joshua Lochner commited on

Improve prediction pipeline
813b772

Joshua Lochner commited on

Merge duplicated training dataclasses
490a61c

Joshua Lochner commited on

Upgrade classifier to transformer-based model
36f7534

Joshua Lochner commited on

Add support for mute action type and remove videos with full action type
1286fe5

Joshua Lochner commited on

Fix conflicting `--no_cuda` argument
09cabec

Joshua Lochner commited on

Remove redundant calls to change device
8981122

Joshua Lochner commited on

Move `CATGEGORY_OPTIONS` to shared
fca2a61

Joshua Lochner commited on

Move `seconds_to_time` to shared
bb58e90

Joshua Lochner commited on

Add tokens for each category
4f1ee08

Joshua Lochner commited on

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

Joshua Lochner commited on

Add source code
5fbdd3c

Joshua Lochner commited on