alteryx / alteryx/featuretools
Add method to split EntitySet
Open
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
Assessment
This issue has not been assessed yet.