nni
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning. (by microsoft)
optuna
A hyperparameter optimization framework (by optuna)
nni | optuna | |
---|---|---|
5 | 35 | |
13,991 | 11,798 | |
- | 2.8% | |
1.8 | 9.9 | |
10 months ago | about 18 hours ago | |
Python | Python | |
MIT License | MIT License |
The number of mentions indicates the total number of mentions that we've tracked plus the number of user suggested alternatives.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
nni
Posts with mentions or reviews of nni.
We have used some of these posts to build our list of alternatives
and similar projects. The last one was on 2021-10-04.
- Filter Pruning for PyTorch
-
Automated Machine Learning (AutoML) - 9 Different Ways with Microsoft AI
For a complete tutorial, navigate to this Jupyter Notebook: https://github.com/microsoft/nni/blob/master/examples/notebooks/tabular_data_classification_in_AML.ipynb
-
[D] Efficient ways of choosing number of layers/neurons in a neural network
optuna, hyperopt, nni, plenty of less-known tools too.
-
Top 10 Developer Trends, Sun Oct 18 2020
microsoft / nni
optuna
Posts with mentions or reviews of optuna.
We have used some of these posts to build our list of alternatives
and similar projects. The last one was on 2024-04-06.
-
Optimizing ML Training with Metagradient Descent
Optuna often works fine in this context (even with the memory errors or, with some tuning, with the non-halting runs): https://github.com/optuna/optuna
-
Optuna – A Hyperparameter Optimization Framework
I didn’t even know WandB did hyperparameter optimization, I figured it was a neural network visualizer based on 2 minute papers. Didn’t seem like many alternatives out there to Optuna with TPE + persistence in conditional continuous & discrete spaces.
Anyway, it’s doable to make a multi objective decide_to_prune function with Optuna, here’s an example https://github.com/optuna/optuna/issues/3450#issuecomment-19...
- How to test optimal parameters
- FOSS hyperparameter optimization framework to automate hyperparameter search
-
How did you make that?!
The network configuration process is usually not particularly scientific and mostly relies on empirical observation. For some cases, tools like Optuna can be used to automatically find the optimal parameters. In others, on others, you can look for modern studies which explore the effect of this parameter on performance, such as this study (2022), but these are typically very specific to one particular architecture.
-
[P] We are building a curated list of open source tooling for data-centric AI workflows, looking for contributions.
Keras Tuner, Optuna : https://github.com/optuna/optuna ?
- How to tune more than 2 hyperparameters in Grid Search in Python?
-
Suggestion to optimize algo
I have used OpenTuner, but I don't think it is maintained anymore. I hear tell that Optuna is what to use now, but have not used it myself. https://optuna.org Optuna - A hyperparameter optimization framework
-
Best practices for training PyTorch model
Research the type of model to get an idea of what hyper parameters to use. I recommend using a hyper parameter optimization library like Optuna to get the best configuration
What are some alternatives?
When comparing nni and optuna you can also consider the following projects:
autogluon - Fast and Accurate ML in 3 Lines of Code
Ray - Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
FLAML - A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.
rl-baselines3-zoo - A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
tsflex - Flexible time series feature extraction & processing
hyperopt - Distributed Asynchronous Hyperparameter Optimization in Python