Ejemplos de ColumnSelector.get_feature_names en Python

Ejemplo n.º 1

def test_get_feature_names():
    df = pd.DataFrame({'a': [4, 5, 6], 'b': ['4', '5', '6']})
    transformer = ColumnSelector('a').fit(df)
    assert transformer.get_feature_names() == ['a']

Ejemplo n.º 2

Archivo: test_columnselector.py Proyecto: prateekchandrajha/scikit-lego

def test_get_feature_names():
    df = pd.DataFrame({"a": [4, 5, 6], "b": ["4", "5", "6"]})
    transformer = ColumnSelector("a").fit(df)
    assert transformer.get_feature_names() == ["a"]