Gokart solves reproducibility, task dependencies, constraints of good code, and ease of use for Machine Learning Pipeline.
Documentation for the latest release is hosted on readthedocs.
Here are some good things about gokart.
- The following meta data for each Task is stored separately in a
pkl
file with hash value- task output data
- imported all module versions
- task processing time
- random seed in task
- displayed log
- all parameters set as class variables in the task
- Automatically rerun the pipeline if parameters of Tasks are changed.
- Support GCS and S3 as a data store for intermediate results of Tasks in the pipeline.
- The above output is exchanged between tasks as an intermediate file, which is memory-friendly
pandas.DataFrame
type and column checking during I/O- Directory structure of saved files is automatically determined from structure of script
- Seeds for numpy and random are automatically fixed
- Can code while adhering to SOLID principles as much as possible
- Tasks are locked via redis even if they run in parallel
All the functions above are created for constructing Machine Learning batches. Provides an excellent environment for reproducibility and team development.
Here are some non-goal / downside of the gokart.
- Batch execution in parallel is supported, but parallel and concurrent execution of task in memory.
- Gokart is focused on reproducibility. So, I/O and capacity of data storage can become a bottleneck.
- No support for task visualize.
- Gokart is not an experiment management tool. The management of the execution result is cut out as Thunderbolt.
- Gokart does not recommend writing pipelines in toml, yaml, json, and more. Gokart is preferring to write them in Python.
Within the activated Python environment, use the following command to install gokart.
pip install gokart
A minimal gokart tasks looks something like this:
import gokart
class Example(gokart.TaskOnKart):
def run(self):
self.dump('Hello, world!')
task = Example()
output = gokart.build(task)
print(output)
gokart.build
return the result of dump by gokart.TaskOnKart
. The example will output the following.
Hello, world!
This is an introduction to some of the gokart. There are still more useful features.
Please See Documentation .
Have a good gokart life.
Gokart is a proven product.
- It's actually been used by m3.inc for over 3 years
- Natural Language Processing Competition by Nishika.inc 2nd prize : Solution Repository
gokart is a wrapper for luigi. Thanks to luigi and dependent projects!