BerkeleyLab / inference-engine

A deep learning library for use in high-performance computing applications in modern Fortran
Other
25 stars 6 forks source link

Features: maximize information entropy and variable reporting interval. #170

Closed rouson closed 2 months ago

rouson commented 2 months ago

This pull request adds --report and --bins command-line arguments in the train-cloud-microphysics app:

  1. --report sets the reporting interval, i.e., the number of epochs between printing the cost function to the screen and writing it to the cost.plt file.
  2. --bins sets the number of phase-space hypercubes across the range of each of the 5 output variables. Only one data point per bin retained for using in training a neural network. This choice maximizes the information entropy for a given retained training data set size.