diff --git a/Classification/Digit_Recognition/train_digits_hog.py b/Classification/Digit_Recognition/train_digits_hog.py index d45172c17..c9fbaadc5 100644 --- a/Classification/Digit_Recognition/train_digits_hog.py +++ b/Classification/Digit_Recognition/train_digits_hog.py @@ -17,7 +17,6 @@ data = [] labels = [] - for imagePath in paths.list_images("dataset\\training"): digit = imagePath.split("\\")[-2] @@ -27,6 +26,9 @@ data.append(H) labels.append(digit) +data_arr = asarray(data) +pixels = data_arr.flatten().reshape(6000, 784) + print("[INFO] training classifier...") model = RandomForestClassifier(n_estimators=30, random_state=42) @@ -54,5 +56,8 @@ act_res.append(digit_val) pred_res.append(pred) +print(classification_report(act_res, pred_res)) + + print(classification_report(act_res, pred_res))