Trane is a software package that automatically generates problems for temporal datasets and produces labels for supervised learning. Its goal is to streamline the machine learning problem-solving process.
Install
Install Trane using pip:
python -m pip install trane
Usage
Here's a quick demonstration of Trane in action:
import trane
data, metadata = trane.load_airbnb()
entity_columns = ["location"]
window_size = "2d"
problem_generator = trane.ProblemGenerator(
metadata=metadata,
window_size=window_size,
entity_columns=entity_columns
)
problems = problem_generator.generate()
print(f'Generated {len(problems)} problems.')
print(problems[108])
print(problems[108].create_target_values(data).head(5))
Output:
Generated 168 problems.
For each <location> predict the majority <rating> in all related records in the next 2 days.
location time target
0 London 2021-01-01 5
1 London 2021-01-03 4
2 London 2021-01-05 5
3 London 2021-01-07 4
4 London 2021-01-09 5
Community
- Questions or Issues? Create a GitHub issue.
- Want to Chat? Join our Slack community.
Cite Trane
If you find Trane beneficial, consider citing our paper:
Ben Schreck, Kalyan Veeramachaneni. What Would a Data Scientist Ask? Automatically Formulating and Solving Predictive Problems. IEEE DSAA 2016, 440-451.
BibTeX entry:
@inproceedings{schreck2016would,
title={What Would a Data Scientist Ask? Automatically Formulating and Solving Predictive Problems},
author={Schreck, Benjamin and Veeramachaneni, Kalyan},
booktitle={Data Science and Advanced Analytics (DSAA), 2016 IEEE International Conference on},
pages={440--451},
year={2016},
organization={IEEE}
}
2 maintainers
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file trane-0.7.0.tar.gz.
File metadata
- Download URL: trane-0.7.0.tar.gz
- Upload date:
- Size: 4.4 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/4.0.2 CPython/3.11.6
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ba0abd4d191f40045cdd7f7d548e27a2b7f953c4795a91bdb8387d8729f45608
|
|
| MD5 |
b8518bbee81cb641105f85f8af8fbeee
|
|
| BLAKE2b-256 |
daf65a2b3798c066a6a16c53617fbfb5f977d1711a4c0310c67402b494b35366
|
File details
Details for the file trane-0.7.0-py3-none-any.whl.
File metadata
- Download URL: trane-0.7.0-py3-none-any.whl
- Upload date:
- Size: 4.4 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/4.0.2 CPython/3.11.6
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6d85f8b3b7dfb41135abebd7a75824623d1b95c325fc45af4d745ad510f6c44e
|
|
| MD5 |
97ae78c5c68b5c4c3ebaefc8cbb170d4
|
|
| BLAKE2b-256 |
17d3ec6998318e339d96f575dc69a9f1ebf4d8930dec0d1e848761c63f85d024
|
