功能工具:跳过目标功能

2024-09-28 23:45:47 发布

您现在位置:Python中文网/ 问答频道 /正文

使用Featuretools时是否可以跳过目标功能?例如,考虑虹膜数据集

data = pd.read_csv('https://gist.githubusercontent.com/curran/a08a1080b88344b0c8a7/raw/639388c2cbc2120a14dcf466e85730eb8be498bb/iris.csv')
target = "species"
data[target] = data[target].map({'setosa':0, 'versicolor':1, 'virginica':2})
# Make an entityset and add the entity
es = ft.EntitySet(id='iris dataset')
es.entity_from_dataframe(entity_id='data', dataframe=data,
                         make_index=True, index='index')

# Run deep feature synthesis with transformation primitives
feature_matrix, feature_defs = ft.dfs(entityset=es, target_entity='data',
                                      trans_primitives=['add_numeric', 'multiply_numeric'])

生成的feature_matrix包含无用的功能,例如sepal_width + species。我怎样才能移除它们


Tags: csv功能addidiristargetdataframedata
1条回答
网友
1楼 · 发布于 2024-09-28 23:45:47

您可以在DFS中使用ignore_variables忽略目标功能

feature_matrix, feature_defs = ft.dfs(
    entityset=es,
    target_entity='data',
    trans_primitives=['add_numeric', 'multiply_numeric'],
    ignore_variables={'data': ['species']},
)

相关问题 更多 >