Merge pull request #129 from aadityasinha-dotcom/GammaRegressor

Added Gamma Regressor
blobcity · Oct 14, 2021 · fbf68d7 · fbf68d7
2 parents fef12f2 + 431b4c6
commit fbf68d7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/blobcity/config/regressor_config.py b/blobcity/config/regressor_config.py
@@ -226,4 +226,12 @@ class regressor_config:
                 "verbose":{'bool':[False]},
             }
         ],
+        "GammaRegressor":[
+            linear_model.GammaRegressor,
+            {
+                "alpha":{'float':[1e-4,1.0]},
+                "max_iter":{'int':[1000,10000]},
+                "tol":{'float':[1e-3,0.1]},
+            }
+        ]
     }