Python TPOT.dt_feature_selection Beispiele

Beispiel #1

Datei: tests.py Projekt: rasbt/tpot

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)

Beispiel #2

Datei: tests.py Projekt: nikhilmitrax/tpot

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)