Есть ли функция Keras для получения параметров компиляции?

Question

Зная, что я могу получить информацию о слоях из уже построенной модели с помощью:

model.layers[index].get_config()

Есть ли способ получить компиляцию информации о модели ? Такие как используемая функция потерь, оптимизатор, метрики и т. Д. c.

Answer 1

Из исходного кода , который сохраняет модель - _serialize_model(), есть:

h5dict['training_config'] = json.dumps({
                'optimizer_config': {
                    'class_name': model.optimizer.__class__.__name__,
                    'config': model.optimizer.get_config()
                },
                'loss': model.loss,
                'metrics': model._compile_metrics,
                'weighted_metrics': model._compile_weighted_metrics,
                'sample_weight_mode': model.sample_weight_mode,
                'loss_weights': model.loss_weights,
            }, default=get_json_type).encode('utf8')

Затем можно использовать:

loss = model.loss
optimizer_class_name = model.optimizer.__class__.__name__
optimizer_config = model.optimizer.get_config()
metrics = model._compile_metrics
weighted_metrics = model._compile_weighted_metrics
sample_weight_mode = model.sample_weight_mode
loss_weights = model.loss_weights