Skip to content

PyTorch

Gustavo Rosa edited this page Jan 16, 2020 · 2 revisions

Opytimizer is easily integrated with other frameworks. One can integrate it with the PyTorch package to work with machine learning and meta-heuristic optimization. We offer the following the pre-implemented scripts:

  • cnn.py: Optimize hyperparameters from a Convolutional Neural Network;
  • enhanced_neural_network.py: Optimize hyperparameters from a deep Neural Network;
  • linear_regression.py: Optimize hyperparameters from a Linear Regression;
  • logistic_regression.py: Optimize hyperparameters from a Logistic Regression;
  • lstm.py: Optimize hyperparameters from a Long Short-Term Memory network;
  • neural_network.py: Optimize hyperparameters from a shallow Neural Network.
Clone this wiki locally