Обновить

Комментарии 2

А просто Email в качестве ключа поиска поддерживается?

Да, в конце статьи приведен такой пример

Рассмотрим пример: у нас есть размеченный датасет, содержащий адреса электронной почты и пол их владельцев. Мы хотим создать модель, которая предсказывает пол по email. Вот как можно искать новые признаки с помощью объекта enricher:

enricher = FeaturesEnricher(
    search_keys = {'email': SearchKey.EMAIL}
)

enricher.fit(X_train, y_train, calculate_metrics=True)

X_train содержит признак — адрес электронной почты, а y_train — информацию о поле в тренировочном наборе.

Зарегистрируйтесь на Хабре, чтобы оставить комментарий

Публикации