diff --git a/src/LearningMethods/GradientDescent.h b/src/LearningMethods/GradientDescent.h index 1c7c54e515816b9a5dafd0c12318692cc8cdf329..b072f7ce07e0a158b0fb82ade5590c3bd0a94617 100644 --- a/src/LearningMethods/GradientDescent.h +++ b/src/LearningMethods/GradientDescent.h @@ -35,7 +35,15 @@ namespace lib4neuro { * Number of iterations to reset step size to tolerance/10.0 */ size_t restart_frequency; + + /** + * + */ size_t batch; + + /** + * + */ size_t iter_max; /**