Automatically formulating machine learning tasks for temporal datasets
Trane is a software package for automatically generating prediction problems and generating labels for supervised learning. Trane is a system designed to advance the automation of the machine learning problem solving pipeline.
Install
To install Trane, run the following command:
python -m pip install trane
Example
Below is an example of using Trane:
import trane
data = trane.datasets.load_covid()
table_meta = trane.datasets.load_covid_metadata()
entity_col = "Country/Region"
window_size = "2d"
minimum_data = "2020-01-22"
maximum_data = "2020-03-29"
cutoff_strategy = trane.CutoffStrategy(
entity_col=entity_col,
window_size=window_size,
minimum_data=minimum_data,
maximum_data=maximum_data,
)
time_col = "Date"
problem_generator = trane.PredictionProblemGenerator(
df=data,
entity_col=entity_col,
time_col=time_col,
cutoff_strategy=cutoff_strategy,
table_meta=table_meta,
)
problems = problem_generator.generate(data, generate_thresholds=True)
Citing Trane
If you use Trane, please consider citing the following 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.5.0.tar.gz.
File metadata
- Download URL: trane-0.5.0.tar.gz
- Upload date:
- Size: 758.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/4.0.2 CPython/3.11.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0f6cc5933d4617e602a663287c6d188a9a8d8ccaeb740b872c9094376581c8d1
|
|
| MD5 |
f1fd1681fe7563ca02494911ee468767
|
|
| BLAKE2b-256 |
b6c870afae64953361250d8c47e0a79975bdbfa97b8dc6b2f83ea9cad49fc501
|
File details
Details for the file trane-0.5.0-py3-none-any.whl.
File metadata
- Download URL: trane-0.5.0-py3-none-any.whl
- Upload date:
- Size: 782.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/4.0.2 CPython/3.11.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
80a8c55b11d309adeb4788319ae428d940f98b0011318336520003475dc711f9
|
|
| MD5 |
e6f574a60973a2b704c32643cb55f182
|
|
| BLAKE2b-256 |
c4acdf4fe3e2bf3571af7b00fd4127e30c01116d596e9112a31a66eef0b7bd02
|
