Skip to content

Commit a0e0a82

Browse files
author
whyboris
committed
tensorboard start
1 parent b92e2b6 commit a0e0a82

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

tensorboard.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
print('TensorBoard demonstration')
2+
3+
from hack import hack
4+
hack()
5+
6+
import keras
7+
from keras import layers
8+
from keras.datasets import imdb
9+
from keras.preprocessing import sequence
10+
11+
max_features = 2000
12+
max_len = 500
13+
14+
(x_train, y_train), (x_test, y_test) = imdb.load_data(num_words = max_features)
15+
x_train = sequence.pad_sequences(x_train, maxlen=max_len)
16+
x_tset = sequence.pad_sequences(x_test, maxlen=max_len)
17+
18+
model = keras.Sequential()
19+
model.add(layers.Embedding(max_features, 128, input_length=max_len, name='embed'))
20+
model.add(layers.Conv1D(32, 7, activation='relu'))
21+
model.add(layers.MaxPool1D(5))
22+
model.add(layers.Conv1D(32, 7, activation='relu'))
23+
model.add(layers.GlobalMaxPooling1D())
24+
model.add(layers.Dense(1))
25+
26+
model.summary()
27+
28+
model.compile(optimiser='rmsprop', loss='binary_crossentropy', metrics=['acc'])
29+
30+

0 commit comments

Comments
 (0)