![]() |
(git:b77b4be)
|
Neural Network implementation. More...
Functions/Subroutines | |
subroutine, public | pao_ml_nn_predict (pao, ikind, descriptor, output, variance) |
Uses neural network to make a prediction. | |
subroutine, public | pao_ml_nn_gradient (pao, ikind, descriptor, outer_deriv, gradient) |
Calculate gradient of neural network. | |
subroutine, public | pao_ml_nn_train (pao) |
Trains the neural network on given training points. | |
Neural Network implementation.
subroutine, public pao_ml_neuralnet::pao_ml_nn_predict | ( | type(pao_env_type), pointer | pao, |
integer, intent(in) | ikind, | ||
real(dp), dimension(:), intent(in) | descriptor, | ||
real(dp), dimension(:), intent(out) | output, | ||
real(dp), intent(out) | variance | ||
) |
Uses neural network to make a prediction.
pao | ... |
ikind | ... |
descriptor | ... |
output | ... |
variance | ... |
Definition at line 43 of file pao_ml_neuralnet.F.
subroutine, public pao_ml_neuralnet::pao_ml_nn_gradient | ( | type(pao_env_type), pointer | pao, |
integer, intent(in) | ikind, | ||
real(dp), dimension(:), intent(in), target | descriptor, | ||
real(dp), dimension(:), intent(in) | outer_deriv, | ||
real(dp), dimension(:), intent(out) | gradient | ||
) |
Calculate gradient of neural network.
pao | ... |
ikind | ... |
descriptor | ... |
outer_deriv | ... |
gradient | ... |
Definition at line 67 of file pao_ml_neuralnet.F.
subroutine, public pao_ml_neuralnet::pao_ml_nn_train | ( | type(pao_env_type), pointer | pao | ) |
Trains the neural network on given training points.
pao | ... |
Definition at line 120 of file pao_ml_neuralnet.F.