Вычислять будущие функции с помощью featuretools - PullRequest
Новая
       17

Вычислять будущие функции с помощью featuretools

2 голосов
/ 21 апреля 2019

Я пытаюсь использовать featuretools, чтобы сгенерировать матрицу объектов для обучения прошлых данных и прогнозирования некоторых будущих данных.Итак, это мои настройки: 

import featuretools as ft
import pandas as pd

df_hotel = pd.DataFrame({
    'hotel_id': [1, 2],
})

df_bookings = pd.DataFrame({
    'bookings_id': [1, 2, 3, 4, 5, 6, 7, 8],
    'time': [1, 2, 3, 4, 1, 2, 3, 4],
    'hotel_id': [1, 1, 1, 1, 2, 2, 2, 2],
    'bookings': [1, 2, 3, 4, 5, 6, 7, 8]
})

es = ft.EntitySet()

es = es.entity_from_dataframe(
    entity_id='c',
    dataframe = df_bookings,
    index='bookings_id',
    time_index='time'
)

es = es.entity_from_dataframe(
    entity_id='hotels',
    dataframe=df_hotel,
    index='hotel_id'
)

es = es.add_relationship(
    ft.Relationship(
        es['hotels']['hotel_id'],
        es['bookings']['hotel_id'],
    )
)

И я генерирую матрицу объектов следующим образом: 

feature_matrix, feature_defs = ft.dfs(
    entityset=es,
    target_entity='bookings',
    cutoff_time=3,
    agg_primitives=["mean"]
)
feature_matrix

Однако это дает мне две строки (где время равно 4, после отсечки) где все значения являются NAN.Желаемое поведение также заключается в заполнении значений этих строк (но вычисление агрегации только на основе прошлых данных).Это возможно с featuretools?

...