diff --git a/Makefile b/Makefile index 7c9c61b..40eec09 100644 --- a/Makefile +++ b/Makefile @@ -9,7 +9,7 @@ pylint: *.py pylint -f parseable -j 4 *.py test-shapley: - $(TIME) python qii.py -m shapley final.csv + $(TIME) python qii.py -m shapley final.csv --show test: $(TIME) python qii.py -m average-unary-individual final.csv diff --git a/ml_util.py b/ml_util.py index 758ab1e..bb748cf 100644 --- a/ml_util.py +++ b/ml_util.py @@ -279,7 +279,7 @@ def get_arguments(): parser.add_argument('--max_depth', type=int, default=2, help='Max depth for decision trees and forests') parser.add_argument('--n_estimators', type=int, default=20, help='Number of trees for decision forests') - parser.add_argument('--seed', type=int, default=None, help='Random seed, auto seeded if not specified', type=int) + parser.add_argument('--seed', default=None, help='Random seed, auto seeded if not specified', type=int) parser.add_argument('-i', '--individual', default=0, type=int, help='Index for Individualized Transparency Report') parser.add_argument('-r', '--record-counterfactuals', action='store_true', help='Store counterfactual pairs for causal analysis') diff --git a/qii_lib.py b/qii_lib.py index 26c8c83..85897d5 100644 --- a/qii_lib.py +++ b/qii_lib.py @@ -3,7 +3,7 @@ import pandas as pd import numpy -RECORD_COUNTERFACTUALS = True +RECORD_COUNTERFACTUALS = False def intervene(X, features, x0): """ Constant intervention """