jiehou commited on
Commit
eb88a4b
·
1 Parent(s): aff1e76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ def KNN_predict(train_features, train_labels, test_feature, K):
110
  ### get frequency of classes
111
  class_freq = {}
112
  for i in range(0,10):
113
- class_freq['digit '+str(i)] = float(major_class.count(i)) / len(major_class)
114
 
115
  ### get neighbor images and save to local
116
  neighbor_imgs =np.array(neighbor_imgs)
 
110
  ### get frequency of classes
111
  class_freq = {}
112
  for i in range(0,10):
113
+ class_freq['Digit '+str(i)] = float(major_class.count(i)) / len(major_class)
114
 
115
  ### get neighbor images and save to local
116
  neighbor_imgs =np.array(neighbor_imgs)