def test_DFS_numpy_sets(self):
self.runAlgorithm(ps.DFS(ps.NumpySetRepresentation), "DFS numpyset",
self.result, self.qualities, self.task)
def test_DFS_set(self):
self.runAlgorithm(ps.DFS(ps.SetRepresentation), "DFS set", self.result,
self.qualities, self.task)
def test_DFS_sum(self):
self.task.qf = ps.CountCallsInterestingMeasure(
ps.StandardQFNumeric(self.task.qf.a, False, 'sum'))
self.runAlgorithm(ps.DFS(ps.BitSetRepresentation), "DFS sum",
self.result, self.qualities, self.task)
np.random.seed(1111)
target_variables = np.random.randint(low=0, high=2, size=1000)
target_estimates = np.random.uniform(size=1000)
target = ps.PredictionTarget(target_variables, target_estimates, roc_auc_score)
searchSpace_Nominal = ps.create_nominal_selectors(data, ignore=['credit_amount'])
searchSpace_Numeric = [] #ps.create_numeric_selectors(data, ignore=['credit_amount'], nbins=10)
searchSpace = searchSpace_Nominal + searchSpace_Numeric
task = ps.SubgroupDiscoveryTask(data, target, searchSpace, result_set_size=10, depth=5, qf=ps.CountCallsInterestingMeasure(ps.PredictionQFNumeric(1, False)))
resultBS = ps.BeamSearch().execute(task)
resultA = ps.Apriori(use_numba=False).execute(task)
resultA_numba = ps.Apriori(use_numba=True).execute(task)
resultSimpleDFS = ps.SimpleDFS().execute(task)
resultDFS = ps.DFS(ps.BitSetRepresentation).execute(task)
resultDFS.to_dataframe()
############################################################
## Toy example using the default eval to generate answers ##
############################################################
np.random.seed(1111)
eval_dict = {"AUPRC(ish)":average_precision_score}
target_variables = np.random.randint(low=0, high=2, size=1000)
target_estimates = np.random.uniform(size=1000)
target = ps.PredictionTarget(target_variables, target_estimates, eval_dict=eval_dict)
target = ps.PredictionTarget(target_variables, target_estimates)
def test_DFS(self):
self.task.qf = ps.CountCallsInterestingMeasure(ps.PredictionQFNumeric(self.task.qf.a, False))
self.runAlgorithm(ps.DFS(ps.BitSetRepresentation), "DFS", self.DFSresult, self.DFSqualities, self.task)