#NEWLINE# def test_fm_regression( ) : #NEWLINE# #INDENT# ( w0 , w , V , y , X ) = get_test_problem ( ) #NEWLINE# fm = mcmc . FMRegression ( n_iter = 1000 , rank = 2 , init_stdev = 0.1 ) #NEWLINE# y_pred = fm . fit_predict ( X , y , X ) #NEWLINE# assert ( metrics . r2_score ( y_pred , y ) > 0.99 )