-
Keras-TCN:应用于Keras的时间卷积网
资源介绍
Keras TCN
与所有主要/最新的Tensorflow版本(从1.14到2.4.0+)兼容。
pip install keras-tcn
Keras时间卷积网络。 []
为什么选择时间卷积网络?
与具有相同容量的循环体系结构相比,TCN具有更长的内存。
在各种任务(序列MNIST,添加问题,复制内存,字级PTB ...)上,其性能始终优于LSTM / GRU体系结构。
平行度,灵活的接收场大小,稳定的梯度,训练所需的低内存,可变长度的输入...
放大的因果卷积层堆栈的可视化(Wavenet,2016)
API
通常的方法是导入TCN层,并在Keras模型中使用它。 下面提供了一个回归任务的示例(对于其他示例,请参阅tasks/ ):
from tensorflow . keras . layers import Dense
from tensorflow . keras import Input , Model
from tcn import TCN , tcn_full_summary
batch_size , timesteps , input_dim = No