Joshua Lochner commited on
Commit
3f7ce4e
1 Parent(s): 5dd37ab

Remove unnecessary function call

Browse files
Files changed (1) hide show
  1. src/evaluate.py +1 -5
src/evaluate.py CHANGED
@@ -200,11 +200,7 @@ def main():
200
 
201
  # Make predictions
202
  predictions = predict(video_id, model, tokenizer,
203
- segmentation_args, words)
204
-
205
- # Filter predictions
206
- predictions = filter_predictions(
207
- predictions, classifier, vectorizer, classifier_args)
208
 
209
  labelled_words = add_labels_to_words(words, sponsor_segments)
210
  met = calculate_metrics(labelled_words, predictions)
 
200
 
201
  # Make predictions
202
  predictions = predict(video_id, model, tokenizer,
203
+ segmentation_args, words, classifier_args)
 
 
 
 
204
 
205
  labelled_words = add_labels_to_words(words, sponsor_segments)
206
  met = calculate_metrics(labelled_words, predictions)