使用FeatureTools计算未来的功能

时间:2019-04-21 08:27:21

标签: python pandas feature-extraction featuretools

我正在尝试使用功能工具生成功能矩阵,以训练过去的数据并预测一些将来的数据。这是我的设置:

e

然后我生成一个特征矩阵,如下所示:

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'],
    )
)

但是,这给我的是两行(在缩减后时间为4),所有值均为NAN。期望的行为是也填充这些行的值(但仅基于过去的数据来计算聚合)。 Featuretools是否可能?

0 个答案:

没有答案