Как визуализировать структуру нейронной сети при использовании GluonTS DeepAR?

Я использую GluonTS DeepAR для развертывания моей задачи прогнозирования продаж. После того, как я закончу создание DeepAREstimator, я хотел бы напечатать структуру или сводку нейронной сети, чтобы помочь в моих бумажных работах, которые могут предоставить скрытый слой, нейронные числа или активировать функцию и т. Д. Например, функция «резюме ()» в Керас.

Предоставляет ли GluonTS DeepAR такой API?

      from gluonts.model.deepar import DeepAREstimator
from gluonts.mx.trainer import Trainer

estimator = DeepAREstimator(
    prediction_length=30,
    #context_length=30,
    num_layers = 40,
    num_cells = 40,
    cell_type='gru',
    #lags_seq = 10,
    #time_features = 15,
    #use_feat_static_cat=True,
    use_feat_dynamic_real=True,
    #num_parallel_samples=100,
    #cardinality=[2,1],
    freq="D",
    #validation_sampler = 30,
    trainer=Trainer(ctx="cpu", epochs=5, learning_rate=1e-3,) #batch_size=400,num_batches_per_epoch=200,)
)
predictor = estimator.train(training_data=train_data)

0 ответов

Другие вопросы по тегам