def test_df_feature_selection(): tpot_obj = TPOT() top_10_feature_pairs = ['00002', '00013', '00020', '00021', '00026', '00042', '00043', '00058', '00061', 'class', 'group', 'guess'] assert np.array_equal(tpot_obj.dt_feature_selection(training_testing_data, 10).columns.values, top_10_feature_pairs)
def test_df_feature_selection(): tpot_obj = TPOT() top_10_feature_pairs = [ '00002', '00013', '00020', '00021', '00026', '00042', '00043', '00058', '00061', 'class', 'group', 'guess' ] assert np.array_equal( tpot_obj.dt_feature_selection(training_testing_data, 10).columns.values, top_10_feature_pairs)