当前位置: 首页>>代码示例>>TypeScript>>正文


TypeScript train.momentum方法代码示例

本文整理汇总了TypeScript中deeplearn.train.momentum方法的典型用法代码示例。如果您正苦于以下问题:TypeScript train.momentum方法的具体用法?TypeScript train.momentum怎么用?TypeScript train.momentum使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在deeplearn.train的用法示例。


在下文中一共展示了train.momentum方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的TypeScript代码示例。

示例1: createOptimizer

 private createOptimizer() {
   switch (this.selectedOptimizerName) {
     case 'sgd': {
       return dl.train.sgd(this.learningRate);
     }
     case 'momentum': {
       return dl.train.momentum(+this.learningRate, +this.momentum);
     }
     case 'rmsprop': {
       return new dl.RMSPropOptimizer(+this.learningRate, +this.gamma);
     }
     case 'adagrad': {
       return new dl.AdagradOptimizer(+this.learningRate);
     }
     case 'adadelta': {
       return new dl.AdadeltaOptimizer(+this.learningRate, +this.gamma);
     }
     case 'adam': {
       return new dl.AdamOptimizer(
           +this.learningRate, +this.beta1, +this.beta2);
     }
     case 'adamax': {
       return new dl.AdamaxOptimizer(
           +this.learningRate, +this.beta1, +this.beta2);
     }
     default: {
       throw new Error(`Unknown optimizer "${this.selectedOptimizerName}"`);
     }
   }
 }
开发者ID:ScapeQin,项目名称:deeplearnjs,代码行数:30,代码来源:model-builder.ts


注:本文中的deeplearn.train.momentum方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。