#NEWLINE# def test_multi_logistic_probas( ) : #NEWLINE# #INDENT# lr = TfSoftmaxRegression ( epochs = 200 , eta = 0.75 , minibatches = 1 , random_seed = 1 ) #NEWLINE# lr . fit ( X , y ) #NEWLINE# idx = [ 0 , 50 , 149 ] #NEWLINE# y_pred = lr . predict_proba ( X [ idx ] ) #NEWLINE# exp = np . array ( [ [ 0.99 , 0.01 , 0.0 ] , [ 0.01 , 0.89 , 0.1 ] , [ 0.0 , 0.02 , 0.98 ] ] ) #NEWLINE# np . testing . assert_almost_equal ( y_pred , exp , 2 )