Skip to content

Commit 7023144

Browse files
committed
Added svr.pickle
1 parent ae30c9e commit 7023144

File tree

3 files changed

+220
-5
lines changed

3 files changed

+220
-5
lines changed

importance_features.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from absl import flags
77
from sklearn import svm
88
import glob
9-
import tqdm
9+
from tqdm import tqdm
1010
import cPickle
1111

1212
FLAGS = flags.FLAGS
@@ -39,7 +39,7 @@ def get_features_list(include_y):
3939

4040
return features
4141

42-
def get_importance_features_for_article(enc_states, enc_sentences, sent_indices, tokenizer, sent_representations_separate):
42+
def get_importance_features_for_article(enc_states, enc_sentences, sent_indices, sent_representations_separate):
4343
abs_sent_indices = sent_indices
4444
rel_sent_indices_0_to_10 = get_relative_sent_indices(sent_indices)
4545
sent_lens = get_sent_lens(enc_sentences)

0 commit comments

Comments
 (0)