def test_gp_new_generation(): """Assert that the gp_generation count gets incremented when _gp_new_generation is called""" tpot_obj = TPOT() tpot_obj.pbar = tqdm(total=1, disable=True) assert (tpot_obj.gp_generation == 0) # Since _gp_new_generation is a decorator, and we dont want to run a full # fit(), decorate a dummy function and then call the dummy function. @_gp_new_generation def dummy_function(self, foo): pass dummy_function(tpot_obj, None) assert (tpot_obj.gp_generation == 1)
def test_gp_new_generation(): """Assert that the gp_generation count gets incremented when _gp_new_generation is called""" tpot_obj = TPOT() tpot_obj.pbar = tqdm(total=1, disable=True) assert(tpot_obj.gp_generation == 0) # Since _gp_new_generation is a decorator, and we dont want to run a full # fit(), decorate a dummy function and then call the dummy function. @_gp_new_generation def dummy_function(self, foo): pass dummy_function(tpot_obj, None) assert(tpot_obj.gp_generation == 1)
def test_score_2(): """Ensure that the TPOT score function outputs a known score for a fixed pipeline""" tpot_obj = TPOT() tpot_obj._training_classes = training_classes tpot_obj._training_features = training_features tpot_obj.pbar = tqdm(total=1, disable=True) known_score = 0.981993770448 # Assumes use of the TPOT balanced_accuracy function # Reify pipeline with known score tpot_obj._optimized_pipeline = creator.Individual.\ from_string('_logistic_regression(input_df, 1.0, 0, True)', tpot_obj._pset) # Get score from TPOT score = tpot_obj.score(testing_features, testing_classes) # http://stackoverflow.com/questions/5595425/ def isclose(a, b, rel_tol=1e-09, abs_tol=0.0): return abs(a-b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol) assert isclose(known_score, score)
def test_score_2(): """Assert that the TPOT score function outputs a known score for a fixed pipeline""" tpot_obj = TPOT() tpot_obj.pbar = tqdm(total=1, disable=True) known_score = 0.986318199045 # Assumes use of the TPOT balanced_accuracy function # Reify pipeline with known score tpot_obj._optimized_pipeline = creator.Individual.\ from_string('RandomForestClassifier(input_matrix)', tpot_obj._pset) tpot_obj._fitted_pipeline = tpot_obj._toolbox.compile(expr=tpot_obj._optimized_pipeline) tpot_obj._fitted_pipeline.fit(training_features, training_classes) # Get score from TPOT score = tpot_obj.score(testing_features, testing_classes) # http://stackoverflow.com/questions/5595425/ def isclose(a, b, rel_tol=1e-09, abs_tol=0.0): return abs(a - b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol) assert isclose(known_score, score)
def test_score_2(): """Ensure that the TPOT score function outputs a known score for a fixed pipeline""" tpot_obj = TPOT() tpot_obj._training_classes = training_classes tpot_obj._training_features = training_features tpot_obj.pbar = tqdm(total=1, disable=True) known_score = 0.981993770448 # Assumes use of the TPOT balanced_accuracy function # Reify pipeline with known score tpot_obj._optimized_pipeline = creator.Individual.\ from_string('_logistic_regression(input_df, 1.0, 0, True)', tpot_obj._pset) # Get score from TPOT score = tpot_obj.score(testing_features, testing_classes) # http://stackoverflow.com/questions/5595425/ def isclose(a, b, rel_tol=1e-09, abs_tol=0.0): return abs(a - b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol) assert isclose(known_score, score)
def test_score_2(): """Assert that the TPOT score function outputs a known score for a fixed pipeline""" tpot_obj = TPOT() tpot_obj.pbar = tqdm(total=1, disable=True) known_score = 0.986318199045 # Assumes use of the TPOT balanced_accuracy function # Reify pipeline with known score tpot_obj._optimized_pipeline = creator.Individual.\ from_string('RandomForestClassifier(input_matrix)', tpot_obj._pset) tpot_obj._fitted_pipeline = tpot_obj._toolbox.compile( expr=tpot_obj._optimized_pipeline) tpot_obj._fitted_pipeline.fit(training_features, training_classes) # Get score from TPOT score = tpot_obj.score(testing_features, testing_classes) # http://stackoverflow.com/questions/5595425/ def isclose(a, b, rel_tol=1e-09, abs_tol=0.0): return abs(a - b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol) assert isclose(known_score, score)