Merge pull request #51 from aadityasinha-dotcom/radiusNeighborsClassi…

…fiers Added RadiusNeighborsClassifier
blobcity · Oct 11, 2021 · a9de9ba · a9de9ba
2 parents beac2c3 + 7667e56
commit a9de9ba
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/blobcity/config/classifier_config.py b/blobcity/config/classifier_config.py
@@ -134,6 +134,18 @@ class classifier_config:
                 'booster':{'str':['gbtree', 'gblinear','dart']}
             }
         ],
+        "radius":[
+            neighbors.RadiusNeighborsClassifier,
+            {
+                "radius":{'float':[1.0,10.0]},
+                "weights":{'str':['uniform','distance']},
+                "algorithm":{'str':['auto', 'ball_tree', 'kd_tree', 'brute']},
+                "p":{'int':[1,2]},
+                "metric":{'str':['euclidean', 'manhattan', 'chebyshev', 'minkowski']},
+                "leaf_size":{'int':[10,50]},
+                "outlier_label":{'str':['most_frequent']}
+            }
+        ]
          "bernoullinb":[
             naive_bayes.BernoulliNB,
             {
@@ -183,3 +195,4 @@ class classifier_config:
             }
         ],
     }
+