diff --git a/fsvm/_fuzzy_svc.py b/fsvm/_fuzzy_svc.py index f4a43b9..4a3f51b 100644 --- a/fsvm/_fuzzy_svc.py +++ b/fsvm/_fuzzy_svc.py @@ -49,7 +49,7 @@ class FuzzySVC(ClassifierMixin, BaseEstimator): median, which minimizes the sum of L1 distances. membership_decay : {'exponential', 'linear'} or callable, \ - default='exponential' + default='linear' Method to compute the decay function for membership as in [1]_. If a callable is passed, it should take the output of `distance_metric` method and return the final membership degree in the interval [0, 1]. @@ -229,7 +229,7 @@ class FuzzySVC(ClassifierMixin, BaseEstimator): 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 1, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, - 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, 2, 2, + 2, 2, 2, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2]) """ @@ -253,7 +253,7 @@ def __init__( *, distance_metric="centroid", centroid_metric="euclidean", - membership_decay="exponential", + membership_decay="linear", beta=0.1, balanced=True, C=1.0, diff --git a/fsvm/tests/test_fuzzy_svc.py b/fsvm/tests/test_fuzzy_svc.py index 9f39503..5086307 100644 --- a/fsvm/tests/test_fuzzy_svc.py +++ b/fsvm/tests/test_fuzzy_svc.py @@ -30,7 +30,7 @@ def test_smoke_test(data): # Check the default values assert clf.distance_metric == "centroid" - assert clf.membership_decay == "exponential" + assert clf.membership_decay == "linear" assert clf.beta == 0.1 assert clf.balanced is True