()
| 1975 | } |
| 1976 | |
| 1977 | protected getTrainingConfig(): TrainingConfig { |
| 1978 | return { |
| 1979 | loss: this.getLossIdentifiers(), |
| 1980 | metrics: this.getMetricIdentifiers(), |
| 1981 | optimizer_config: { |
| 1982 | class_name: this.optimizer.getClassName(), |
| 1983 | config: this.optimizer.getConfig() |
| 1984 | } as OptimizerSerialization |
| 1985 | }; |
| 1986 | // TODO(cais): Add weight_metrics when they are supported. |
| 1987 | // TODO(cais): Add sample_weight_mode when it's supported. |
| 1988 | // TODO(cais): Add loss_weights when it's supported. |
| 1989 | } |
| 1990 | |
| 1991 | loadTrainingConfig(trainingConfig: TrainingConfig) { |
| 1992 | if (trainingConfig.weighted_metrics != null) { |
no test coverage detected