Create `get_model_tokenizer` helper method for loading model and tokenizer
bce5ce9
Joshua Lochner
commited on
Spaces:
Running
Running
Commit History
Move `CATGEGORY_OPTIONS` to shared
fca2a61
Joshua Lochner
commited on
Move `seconds_to_time` to shared
bb58e90
Joshua Lochner
commited on
Use classifier category if transformer generates unknown category
ad7fc61
Joshua Lochner
commited on
Remove redundant count
a6de017
Joshua Lochner
commited on
Add compatibility for python 3.6+
7781f10
Joshua Lochner
commited on
Remove unused methods and improve caching
9ced7bd
Joshua Lochner
commited on
Use multiclass classifier to filter predictions
25f1183
Joshua Lochner
commited on
Change to multiclass classifier
320a2ba
Joshua Lochner
commited on
Use `itertools.islice` instead of custom slicing
3af0cd0
Joshua Lochner
commited on
Update evaluate.py
b3b69aa
Joshua Lochner
commited on
Add tokens for each category
4f1ee08
Joshua Lochner
commited on
Move `jaccard` to utils
13df84e
Joshua Lochner
commited on
Improve preprocessing
3879103
Joshua Lochner
commited on
Change `extract_segment` to use a binary search
915339e
Joshua Lochner
commited on
Move `jaccard` to utils
b309907
Joshua Lochner
commited on
Format utils code
a197a49
Joshua Lochner
commited on
Improve parallel execution of functions
07690ba
Joshua Lochner
commited on
Allow seeding for evaluations
63f1925
Joshua Lochner
commited on
Temporarily disable filtering of predictions using classifier
69fe24d
Joshua Lochner
commited on
Improve segmentation logic
c2ccf6d
Joshua Lochner
commited on
Add functionality to predict self-promo and interaction reminders
90d1f68
Joshua Lochner
commited on
Add cache system for loading classifiers/vectorizers
4678c9b
Joshua Lochner
commited on
Improve prediction output format
74c1216
Joshua Lochner
commited on
Remove unnecessary function call
3f7ce4e
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