-
Notifications
You must be signed in to change notification settings - Fork 262
Description
(tensorflow) F:\Postgraduate\KaggleLearning\multi-class-text-classification-cnn-rnn-master\multi-class-text-classification-cnn-rnn-master>python predict.py ./t
rained_results_1541818386/ ./data2/samples.csv
D:\Anaconda\anaconda\envs\tensorflow\lib\site-packages\gensim\utils.py:1212: UserWarning: detected Windows; aliasing chunkize to chunkize_serial
warnings.warn("detected Windows; aliasing chunkize to chunkize_serial")
Traceback (most recent call last):
File "predict.py", line 141, in
predict_unseen_data()
File "predict.py", line 68, in predict_unseen_data
x_, y_, df = load_test_data(test_file, labels)
File "predict.py", line 43, in load_test_data
y_ = df[select[1]].apply(lambda x: label_dict[x]).tolist()
File "D:\Anaconda\anaconda\envs\tensorflow\lib\site-packages\pandas\core\series.py", line 3194, in apply
mapped = lib.map_infer(values, f, convert=convert_dtype)
File "pandas/_libs/src\inference.pyx", line 1472, in pandas.libs.lib.map_infer
File "predict.py", line 43, in
y = df[select[1]].apply(lambda x: label_dict[x]).tolist()
KeyError: 'RECOVERED VEHICLE'