private Map < Integer , Double > predict ( final double [ ] x ) { Map < Integer , Double > result = new HashMap <> ( ) ; for ( int i = 0 ; i < model . weights . length ; i ++ ) { double y = VectorUtils . dotProduct ( x , model . weights [ i ] ) ; y += model . bias [ i ] ; result . put ( i , y ) ; } return result ; }
Key is LabelIndex .