如何从通过深度特征合成创建的feature_def中选择特征

时间:2018-12-20 14:54:22

标签: featuretools

我正在使用深度特征综合来创建新特征。如何从feature_def中选择功能。

例如,我需要选择所有带有字符串“ Age”的功能。 我尝试了以下代码,但出现错误“类型'IdentityFeature'的参数不可迭代”

    feature_matrix, feature_defs = ft.dfs(entityset= es, target_entity= 'titanic', max_depth= 2)
    features = []
    for s in feature_defs:
      if 'Age' in s:
        features.append(s)

1 个答案:

答案 0 :(得分:1)

您需要在功能定义上使用.get_name()方法。例如,

feature_matrix, feature_defs = ft.dfs(entityset= es, target_entity= 'titanic', max_depth= 2)
features = []
for s in feature_defs:
  if 'Age' in s.get_name():
    features.append(s)