Method test_numpy_input

python/seldon/tests/test_vw.py:175–184

(self)

    def test_numpy_input(self):
        t = vw.VWClassifier()
        try:
            X = np.random.randn(6,4)
            y = np.array([1,2,1,1,2,2])
            t.fit(X,y)
            scores = t.predict_proba(X)
            print scores
        finally:
            t.close()

        
if __name__ == '__main__':