NBeatsConfig

class tfts.models.nbeats.NBeatsConfig(stack_types=['trend_block', 'seasonality_block'], nb_blocks_per_stack=3, num_block_layers=4, hidden_size=64, thetas_dims=(4, 8), share_weights_in_stack=False)

Bases: BaseConfig

Inherited-members:

Methods

from_dict(config_dict)
from_json(json_file)
from_pretrained(pretrained_model_name_or_path)
save_pretrained(save_directory)
to_dict()
to_json(json_file)
update(config_dict)

Attributes

attribute_map
model_type