You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

AdamOptimizer.cs 704 B

12345678910111213141516171819202122232425
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Text;
  4. namespace Tensorflow.Train
  5. {
  6. /// <summary>
  7. /// Optimizer that implements the Adam algorithm.
  8. /// http://arxiv.org/abs/1412.6980
  9. /// </summary>
  10. public class AdamOptimizer : Optimizer
  11. {
  12. private float _beta1;
  13. private float _beta2;
  14. private float _epsilon;
  15. public AdamOptimizer(float learning_rate, float beta1 = 0.9f, float beta2 = 0.999f, float epsilon = 1e-8f, bool use_locking = false, string name = "Adam")
  16. : base(learning_rate, use_locking, name)
  17. {
  18. _beta1 = beta1;
  19. _beta2 = beta2;
  20. _epsilon = epsilon;
  21. }
  22. }
  23. }

tensorflow框架的.NET版本,提供了丰富的特性和API,可以借此很方便地在.NET平台下搭建深度学习训练与推理流程。