def test_binary_blocks(): X, Y = toy.generate_blocks(n_samples=10) crf = GridCRF() clf = StructuredPerceptron(problem=crf, max_iter=30) clf.fit(X, Y) Y_pred = clf.predict(X) assert_array_equal(Y, Y_pred)
def test_multinomial_blocks(): X, Y = toy.generate_blocks_multinomial(n_samples=10, noise=0.3, seed=0) crf = GridCRF(n_states=X.shape[-1]) clf = StructuredPerceptron(problem=crf, max_iter=10) clf.fit(X, Y) Y_pred = clf.predict(X) assert_array_equal(Y, Y_pred)