TCNConfig#
- class tfts.models.tcn.TCNConfig(dilation_rates: List[int] = [1, 2, 4, 8], kernel_sizes: List[int] = [2, 2, 2, 2], filters: int = 128, dense_hidden_size: int = 64)[source]#
Bases:
BaseConfigInitializes the configuration for the Temporal Convolutional Network (TCN) model with the specified parameters.
- Parameters:
dilation_rates – List of dilation rates for each layer.
kernel_sizes – List of kernel sizes for each convolutional layer.
filters – The number of filters (channels) in each convolutional layer.
dense_hidden_size – The size of the dense hidden layer.
- 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_mapmodel_type