UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

18 lines (17 loc) 606 B
import { NamedVariableMap } from '../types'; import { Optimizer } from './optimizer'; export declare class RMSPropOptimizer extends Optimizer { protected learningRate: number; private c; private epsilon; private decay; private momentum; private oneMinusDecay; private centered; private accumulatedMeanSquares; private accumulatedMeanGrads; private accumulatedMoments; constructor(learningRate: number, decay?: number, momentum?: number, epsilon?: number, centered?: boolean); applyGradients(variableGradients: NamedVariableMap): void; dispose(): void; }