alteryx / alteryx/featuretools

Add method to split EntitySet

Open
#2,206 3 comments 0 reactions 0 assignees View on GitHub
new feature
Dominant language
Python
Stars
7.7k
Forks
915
PR merge metrics
No merged PRs in 30d

Description

As a developer, I wish I could use Featuretools to split my EntitySet into train and test datasets. Sometimes all of the data will be included when an EntitySet is created, and having the ability to separate the data into independent EntitySets for modeling could be useful.

The relationships defined in the EntitySet could be used to grab all the relevant rows from the related tables.

#### Code Example

```python
train_es, test_es = EntitySet.split(target_dataframe_name="transactions", test_size=0.2, ...)`
```

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.