diff --git a/Chrono/TimePhraseToChrono/PeriodInterval.py b/Chrono/TimePhraseToChrono/PeriodInterval.py index 89bfb3eb..e60791a6 100644 --- a/Chrono/TimePhraseToChrono/PeriodInterval.py +++ b/Chrono/TimePhraseToChrono/PeriodInterval.py @@ -205,6 +205,9 @@ def buildPeriodInterval(s, chrono_id, chrono_list, ref_list, classifier, feats): if(classifier[1] == "NN"): my_class = ChronoKeras.keras_classify(classifier[0], np.array(list(my_features.values()))) #print("Class: " + str(my_class) + " : Start: " + str(abs_Sspan) + " : End: "+ str(abs_Espan)) + elif classifier[1] in ("SVM", "RF"): + feat_array = [int(i) for i in my_features.values()] + my_class = classifier[0].predict([feat_array])[0] else: my_class = classifier[0].classify(my_features) #print("Class: " + str(my_class) + " : Start: " + str(abs_Sspan) + " : End: "+ str(abs_Espan))