add import keras models

vecto-ai · Oct 13, 2018 · d10ac36 · d10ac36
1 parent 48ac33e
commit d10ac36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vecto/benchmarks/relation_extraction/relation_extraction.py b/vecto/benchmarks/relation_extraction/relation_extraction.py
@@ -9,7 +9,7 @@
 from ..base import Benchmark
 import os
 
-# from keras.models import Model
+from keras.models import Model
 # from keras.layers import Input, Dense, Dropout, concatenate
 # from keras.layers import Embedding
 # from keras.layers import Convolution1D, GlobalMaxPooling1D