Передать параметры примитивам агрегации в featuretools - PullRequest
1 голос
/ 01 ноября 2019

Я использую функцию featuretools dfs для генерации моей feature_matrix с использованием примитивов агрегации, связанных со временем, таких как TimeSince (https://docs.featuretools.com/api_reference.html#deep-feature-synthesis)

feature_matrix, feature_defs = ft.dfs(entityset=es,
                                     target_entity="users",
                                     max_depth=2,
                                     agg_primitives=["sum", "max", "min", 
                                                     "mean", "median", "count", 
                                                     "avg_time_between"],
                                     trans_primitives=["day", "year", "month", "weekday", 
                                                       "time_since_previous", "time_since",
                                                      ],
                                     cutoff_time=cutoff_times,
                                     cutoff_time_in_index=True)

Как передать [единицу]параметры аргумента trans_primitives для изменения единицы времени на минуты, часы и т. д .?

1 Ответ

2 голосов
/ 01 ноября 2019

Чтобы передать параметры в примитив агрегации или преобразования, вы можете создать экземпляр примитива с параметрами. Вот как передать параметр unit в примитив TimeSince.

from featuretools.primitives import TimeSince

ft.dfs(
    ...
    trans_primitives=[TimeSince(unit='hours')]
)
...