Как визуализировать структуру нейронной сети при использовании 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)