本文整理汇总了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}"`);
}
}
}