Spaces:
Sleeping
Sleeping
File size: 1,278 Bytes
2db1521 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import tensorflow as tf
import tensorflow.keras.layers as layers
input_shape = (1,54,1)
model = tf.keras.models.Sequential()
model.add(layers.Conv1D(31, 7, activation='relu', input_shape=input_shape[1:]))
model.add(layers.MaxPooling1D(7, data_format='channels_first'))
model.add(layers.Conv1D(31, 7, activation='relu'))
model.add(layers.MaxPooling1D(7, data_format='channels_first'))
model.add(layers.Conv1D(31, 7, activation='relu'))
model.add(layers.MaxPooling1D(7, data_format='channels_first'))
model.add(layers.GRU(7))
model.add(layers.Dense(18))
model.compile(optimizer='sgd', loss='mse')
print("\n===============\n")
x = tf.random.normal(input_shape)
#print("x: ", x)
y = model.evaluate(x)
#print("y: ", y)
model = layers.Conv1D(31, 7, activation='relu', input_shape=input_shape[1:])(x)
print(model.shape)
model = layers.MaxPooling1D(7, data_format='channels_first')(model)
print(model.shape)
model = layers.Conv1D(31, 7, activation='relu')(model)
print(model.shape)
model = layers.MaxPooling1D(7, data_format='channels_first')(model)
print(model.shape)
model = layers.Conv1D(31, 7, activation='relu')(model)
print(model.shape)
model = layers.MaxPooling1D(7, data_format='channels_first')(model)
print(model.shape)
model = layers.GRU(7)(model)
print(model.shape)
|