Skip to content

Commit

Permalink
Change default parameter of SVC model in the GCM module
Browse files Browse the repository at this point in the history
Before, the Support Vector Classifier did not produce probabilities, which are required for different algorithms in the GCM module. This changes the 'probability' parameter to True.

Signed-off-by: Patrick Bloebaum <bloebp@amazon.com>
  • Loading branch information
bloebp committed Nov 16, 2023
1 parent e846851 commit e5848b6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dowhy/gcm/ml/classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ def create_ada_boost_classifier(**kwargs) -> SklearnClassificationModel:


def create_support_vector_classifier(**kwargs) -> SklearnClassificationModel:
return SklearnClassificationModel(SVC(**kwargs))
return SklearnClassificationModel(SVC(**kwargs, probability=True))


def create_knn_classifier(**kwargs) -> SklearnClassificationModel:
Expand Down

0 comments on commit e5848b6

Please sign in to comment.