A wrapper layer for stacking layers horizontally
Project description
A wrapper layer for stacking layers horizontally.
Install
pip install keras-multi-head
Usage
Duplicate Layers
The layer will be duplicated if only a single layer is provided. The layer_num argument controls how many layers will be duplicated eventually.
import keras
from keras_multi_head import MultiHead
model = keras.models.Sequential()
model.add(keras.layers.Embedding(input_dim=100, output_dim=20, name='Embedding'))
model.add(MultiHead(keras.layers.LSTM(units=32), layer_num=5, name='Multi-LSTMs'))
model.add(keras.layers.Flatten(name='Flatten'))
model.add(keras.layers.Dense(units=4, activation='softmax', name='Dense'))
model.build()
model.summary()
Use Multiple-Layers
The first argument could also be a list of layers with different configurations, however, they must have the same output shapes.
import keras
from keras_multi_head import MultiHead
model = keras.models.Sequential()
model.add(keras.layers.Embedding(input_dim=100, output_dim=20, name='Embedding'))
model.add(MultiHead([
keras.layers.Conv1D(filters=32, kernel_size=3, padding='same'),
keras.layers.Conv1D(filters=32, kernel_size=5, padding='same'),
keras.layers.Conv1D(filters=32, kernel_size=7, padding='same'),
], name='Multi-CNNs'))
model.build()
model.summary()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
keras-multi-head-0.3.tar.gz
(3.2 kB
view hashes)