def _test_boston_rbf_compare_with_sklearn(C, nu, gamma): diabetes = datasets.load_boston() clf = NuSVR(kernel='rbf', gamma=gamma, C=C, nu=nu) clf.fit(diabetes.data, diabetes.target) result = clf.score(diabetes.data, diabetes.target) clf = SklearnNuSVR(kernel='rbf', gamma=gamma, C=C, nu=nu) clf.fit(diabetes.data, diabetes.target) expected = clf.score(diabetes.data, diabetes.target) assert result > 0.4 assert abs(result - expected) < 1e-3
def _test_boston_poly_compare_with_sklearn(params): diabetes = datasets.load_boston() clf = NuSVR(kernel='poly', **params) clf.fit(diabetes.data, diabetes.target) result = clf.score(diabetes.data, diabetes.target) clf = SklearnNuSVR(kernel='poly', **params) clf.fit(diabetes.data, diabetes.target) expected = clf.score(diabetes.data, diabetes.target) assert result > 0.5 assert abs(result - expected) < 1e-3
def _test_boston_linear_compare_with_sklearn(C, nu): diabetes = datasets.load_boston() clf = NuSVR(kernel='linear', C=C, nu=nu) clf.fit(diabetes.data, diabetes.target) result = clf.score(diabetes.data, diabetes.target) clf = SklearnNuSVR(kernel='linear', C=C, nu=nu) clf.fit(diabetes.data, diabetes.target) expected = clf.score(diabetes.data, diabetes.target) assert result > 0.5 assert abs(result - expected) < 1e-3
def _test_diabetes_compare_with_sklearn(kernel): diabetes = datasets.load_diabetes() clf_onedal = NuSVR(kernel=kernel, nu=.25, C=10.) clf_onedal.fit(diabetes.data, diabetes.target) result = clf_onedal.score(diabetes.data, diabetes.target) clf_sklearn = SklearnNuSVR(kernel=kernel, nu=.25, C=10.) clf_sklearn.fit(diabetes.data, diabetes.target) expected = clf_sklearn.score(diabetes.data, diabetes.target) assert result > expected - 1e-5 assert_allclose(clf_sklearn.intercept_, clf_onedal.intercept_, atol=1e-4) assert_allclose(clf_sklearn.support_vectors_.shape, clf_sklearn.support_vectors_.shape) assert_allclose(clf_sklearn.dual_coef_, clf_onedal.dual_coef_, atol=1e-2)
def test_diabetes_simple(): diabetes = datasets.load_diabetes() clf = NuSVR(kernel='linear', C=10.) clf.fit(diabetes.data, diabetes.target) assert clf.score(diabetes.data, diabetes.target) > 0.02