skip newsgroup header (keras-team#5585)

satels · Mar 4, 2017 · 65ce238 · 65ce238
1 parent 5b0967a
commit 65ce238
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/examples/pretrained_word_embeddings.py b/examples/pretrained_word_embeddings.py
@@ -66,7 +66,11 @@
                     f = open(fpath)
                 else:
                     f = open(fpath, encoding='latin-1')
-                texts.append(f.read())
+                t = f.read()
+                i = t.find('\n\n')  # skip header
+                if 0 < i:
+                    t = t[i:]
+                texts.append(t)
                 f.close()
                 labels.append(label_id)
 
@@ -141,4 +145,4 @@
 
 # happy learning!
 model.fit(x_train, y_train, validation_data=(x_val, y_val),
-          epochs=2, batch_size=128)
+          epochs=10, batch_size=128)