Hyperparameter Optimization for TensorFlow, Keras and PyTorch
Talos • Key Features • Examples • Install • Support • Docs • Issues • License • Download
Talos radically changes the ordinary Keras, TensorFlow (tf.keras), and PyTorch workflow by fully automating hyperparameter tuning and model evaluation. Talos exposes Keras and TensorFlow (tf.keras) and PyTorch functionality entirely and there is no new syntax or templates to learn.
TL;DR
Talos radically transforms ordinary Keras, TensorFlow (tf.keras), and PyTorch workflows without taking away.
Talos is made for data scientists and data engineers that want to remain in complete control of their TensorFlow (tf.keras) and PyTorch models, but are tired of mindless parameter hopping and confusing optimization solutions that add complexity instead of reducing it. Within minutes, without learning any new syntax, Talos allows you to configure, perform, and evaluate hyperparameter optimization experiments that yield state-of-the-art results across a wide range of prediction tasks. Talos provides the simplest and yet most powerful available method for hyperparameter optimization with TensorFlow (tf.keras) and PyTorch.
Based on what no doubt constitutes a "biased" review (being our own) of more than ~30 hyperparameter tuning and optimization solutions, Talos comes on top in terms of intuitive, easy-to-learn, highly permissive access to critical hyperparameter optimization capabilities. Key features include:
talos.Scan(x, y, model, params).predict(x_test, y_test)
Talos works on Linux, Mac OSX, and Windows systems and can be operated cpu, gpu, and multi-gpu systems.
Get the below code here. More examples further below.
The Simple example below is more than enough for starting to use Talos with any Keras model. Field Report has +2,600 claps on Medium because it's more entertaining.
Simple [1-2 mins]
Concise [~5 mins]
Comprehensive [~10 mins]
Field Report [~15 mins]
For more information on how Talos can help with your Keras, TensorFlow (tf.keras) and PyTorch workflow, visit the User Manual.
You may also want to check out a visualization of the Talos Hyperparameter Tuning workflow.
Stable version:
pip install talos
Daily development version:
pip install git+https://github.com/autonomio/talos
I want to... | Go to... |
---|---|
...troubleshoot | Docs · Wiki · GitHub Issue Tracker |
...report a bug | GitHub Issue Tracker |
...suggest a new feature | GitHub Issue Tracker |
...get support | Stack Overflow |
If you use Talos for published work, please cite:
Autonomio Talos [Computer software]. (2020). Retrieved from http://github.com/autonomio/talos.