File tree Expand file tree Collapse file tree 2 files changed +13
-3
lines changed
out/production/classification/classification Expand file tree Collapse file tree 2 files changed +13
-3
lines changed Original file line number Diff line number Diff line change @@ -171,15 +171,25 @@ private void testClassificationModel() {
171171
172172
173173 }
174-
174+ float min = Float .valueOf (this .predictedTestData [i ][0 ][1 ]);
175+ int tempIndex = 0 ;
175176 for (int j = 0 ; j <this .numberOfClasses ; j ++) {
176- this .sortedProbability [i ][j ] = Integer .valueOf (this .predictedTestData [i ][j ][1 ]);
177+ if (min > Float .valueOf (this .predictedTestData [i ][j ][1 ])) {
178+ min = Float .valueOf (this .predictedTestData [i ][j ][1 ]);
179+ tempIndex = j ;
180+
181+ }
182+
177183
178184 }
185+ this .sortedProbability [i ][0 ] = tempIndex ;
186+ System .out .println (this .sortedProbability [i ][0 ]);
179187
180188
189+ if (i == 5 ) {
190+ break ;
191+ }
181192
182- break ;
183193 //int[] tempProbability = new int[this.numberOfClasses];
184194 /*
185195 for (int j = 0; j < this.numberOfClasses; j++) {
You can’t perform that action at this time.
0 commit comments