@ Override public List < String > predict ( DataSet dataSet ) { int [ ] intRet = predict ( dataSet . getFeatures ( ) ) ; List < String > ret = new ArrayList <> ( ) ; for ( int i : intRet ) { ret . add ( i , dataSet . getLabelName ( i ) ) ; } return ret ; }
Return predicted label names