@ Override public DataSet vectorize ( String text , String label ) { INDArray input = transform ( text ) ; INDArray labelMatrix = FeatureUtil . toOutcomeVector ( labelsSource . indexOf ( label ) , labelsSource . size ( ) ) ; return new DataSet ( input , labelMatrix ) ; }
Vectorizes the passed in text treating it as one document