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

Timestamp:

Jul 26, 2019, 4:13:00 PM (5 years ago)

Author:

Murray Heymann <heymann.murray@…>

Branches:

(u'spielwiese', 'fe61d9c35bf7c61f2b6cbf1b56e25e2f08d536cc')

Children:

55a5abbd0397943b2c98a4492b27ecbc050c20ef

Parents:

746d3db65a7b75cbc29407ee9382332483801f1d

Message:

Reorganize source tree, make test prediction

File:

• machine_learning/model/predictor.py (moved) (moved from machine_learning/common/predictor.py) (2 diffs)

machine_learning/model/predictor.py

@@ -8 +8 @@
 
 # Local imports
-from keyword_vector import vector_distance
+from common.keyword_vector import vector_distance, count_occurances, \
+                                    read_dictionary
+from common.lookuptable import create_table
+from common.constants import KEYWORDS_FILE
 
 
@@ -89 +92 @@
     print(prediction)
 
+    dictionary = read_dictionary(KEYWORDS_FILE)
+    vectors, file_list = create_table(dictionary=dictionary)
+    test_vec = count_occurances("extract.lib", dictionary)
+    predictor.fit(vectors, file_list)
+    prediction = predictor.predict(np.array([test_vec]))
+    print(prediction)
+
 if __name__ == '__main__':
     main()