Changeset f59883 in git for machine_learning/model/predictor.py

Timestamp:

Jul 30, 2019, 5:21:47 PM (4 years ago)

Author:

Murray Heymann <heymann.murray@…>

Branches:

(u'spielwiese', '8e0ad00ce244dfd0756200662572aef8402f13d5')

Children:

da892581a52069935f084604d05c0ecd6d19d5c9

Parents:

de88e1d84f18979811df4c3d1dc68f221a3362ef

Message:

Expand testing

File:

• machine_learning/model/predictor.py (modified) (3 diffs)

machine_learning/model/predictor.py

@@ -120 +120 @@
     print(end - start, "seconds to create_table")
 
-    test_vec = count_occurances("extract.lib", dictionary)
     predictor = HelpPagePredictor()
     predictor.fit(vectors, file_list)
 
     start = time.time()
+    test_vec = count_occurances("extract.lib", dictionary)
     prediction = predictor.predict(np.array([test_vec]))
     end = time.time()
@@ -132 +132 @@
 
     print("prediction for zero vector")
+    start = time.time()
     zerovec = np.zeros(len(dictionary) - 2)
-    start = time.time()
     prediction = predictor.predict(np.array([zerovec]))
     end = time.time()
@@ -147 +147 @@
                 continue
             print("predicting for file", sys.argv[i])
+            start = time.time()
             test_vec = count_occurances(sys.argv[i], dictionary)
-            start = time.time()
             prediction = predictor.predict(np.array([test_vec]))
             end = time.time()