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)
