Skip to content

Using Optuna to optimize hyperparameters for autoencoder model in Pytorch

Notifications You must be signed in to change notification settings

HelenShao/Optuna_Autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

42 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Optuna Autoencoder

Using Optuna to optimize hyperparameters for autoencoder model implemented in Pytorch

Goal of Project: To determine how to accruately represent dark matter halos with reduced number of properties using autoencoder and PCA Finding hidden relationships between the 11 halo properties with symbolic regression using PYSR

Important Features:

  1. Dynamic Autoencoder Architecture - located in architecture.py to allow for optuna optimization of model
  2. Data - Normalized and split into train, valid, and test loaders for training
  3. Training Parameters - how to implement optuna and what parameters are needed (input_size, batch_size, seed, etc)

About

Using Optuna to optimize hyperparameters for autoencoder model in Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages