Выбор функций в Python

Question

Я пытаюсь выполнить выбор функций в Python, используя несколько методов. Первая техника, которую я пытаюсь применить, - это выбор функций, используя только разнообразие функций. Мой код следующий:

def feature_selection_technique(train, test, lbls, technique):
   if technique == "variance":
        sel = VarianceThreshold(threshold=(0.00010 * (1 - .15)))
        model1 = sel1.fit_transform(face_train)
        new_train = model1.transform(train)
        new_test = model1.transform(test)

    return new_train, new_test

На самом деле я хочу вычислить выбранные объекты, используя набор данных поезда, а затем применить его к тестовому набору данных. Кажется, что метод преобразования не может быть отключен в этом случае. Что я могу сделать в этом случае?

Answer 1

Я думаю, что есть проблема в синтаксисе, который вы используете. Смотрите документацию и пример здесь. Правильный синтаксис выглядит следующим образом:

def feature_selection_technique(train, test, lbls, technique):
   if technique == "variance":
        sel = VarianceThreshold(threshold=(0.00010 * (1 - .15)))
        new_train=sel.fit_transform(train)
        new_test = sel.transform(test)

    return new_train, new_test

То есть вы должны инициализировать selзатем подгоните его под тренировочные данные и преобразуйте, а затем преобразуйте тестовые данные.