如何修复Colab上的featurtools类型错误

Question

我在下面的代码中的es.entity_from_dataframe（..）中输入time_index ='date'参数时，ft.dfs（..）会抛出一堆从Type Error（类型错误）开始的错误。我正在使用带有功能工具0.4.1。的google colab。

import pandas as pd

import featuretools as ft

df1 = pd.DataFrame({'df_index' : [1,2,3,4,5],
                 'location':['aust','aust','aust','canada','canada'],
                  'prices':[34,52,46,25,67],
                   'values':[786,345,123,654,841]
                  })

es = ft.EntitySet(id='Transactions')

es.entity_from_dataframe(entity_id='log', 
                         dataframe=df1, 
                         index='df_index',
                         time_index='date'
                        )

es.normalize_entity(base_entity_id='log', new_entity_id='loc', index= 'location' )


fm, features = ft.dfs(entityset=es, target_entity='log',
                      trans_primitives = ['add', 'multiply'],
                      agg_primitives = ['sum', 'mean'],
                      max_depth = 2,
                      verbose = 2
                     )

Answer 1

Colab当前捆绑了Featuretools 0.4.1，我怀疑您正在使用较新的API。我将首先像这样升级featuretools库：

!pip install -U featuretools

此后，您需要使用“运行时->重新启动”菜单重新启动Python进程。

然后，您会看到其他错误，例如：

LookupError: Time index not found in dataframe

但是，我认为这是因为您要引用的date列不存在于DataFrame变量df1中。