1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 | import { AdadeltaOptimizer } from './optimizers/adadelta_optimizer';
|
19 | import { AdagradOptimizer } from './optimizers/adagrad_optimizer';
|
20 | import { AdamOptimizer } from './optimizers/adam_optimizer';
|
21 | import { AdamaxOptimizer } from './optimizers/adamax_optimizer';
|
22 | import { MomentumOptimizer } from './optimizers/momentum_optimizer';
|
23 | import { OptimizerConstructors } from './optimizers/optimizer_constructors';
|
24 | import { RMSPropOptimizer } from './optimizers/rmsprop_optimizer';
|
25 | import { SGDOptimizer } from './optimizers/sgd_optimizer';
|
26 |
|
27 | [MomentumOptimizer, SGDOptimizer, AdadeltaOptimizer, AdagradOptimizer,
|
28 | RMSPropOptimizer, AdamaxOptimizer, AdamOptimizer];
|
29 | export const train = {
|
30 | sgd: OptimizerConstructors.sgd,
|
31 | momentum: OptimizerConstructors.momentum,
|
32 | adadelta: OptimizerConstructors.adadelta,
|
33 | adagrad: OptimizerConstructors.adagrad,
|
34 | rmsprop: OptimizerConstructors.rmsprop,
|
35 | adamax: OptimizerConstructors.adamax,
|
36 | adam: OptimizerConstructors.adam
|
37 | };
|
38 |
|
\ | No newline at end of file |