Skip to content

Commit 8bb5d5d

Browse files
committed
remove teams for class project
1 parent 05410e5 commit 8bb5d5d

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

keras-sign/perceptron.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,10 @@
1010
# logging code
1111
run = wandb.init()
1212
config = run.config
13-
config.team_name = "default"
1413
config.loss = "categorical_crossentropy"
1514
config.optimizer = "adam"
1615
config.epochs = 10
1716

18-
if (config.team_name == 'default'):
19-
raise ValueError("Please set config.team_name to be your team name")
20-
2117
# load data
2218
(X_test, y_test) = signdata.load_test_data()
2319
(X_train, y_train) = signdata.load_train_data()
@@ -36,11 +32,10 @@
3632
# create model
3733
model=Sequential()
3834
model.add(Flatten(input_shape=(img_width, img_height)))
39-
model.add(Dense(100, activation='relu'))
40-
model.add(Dense(num_classes, activation='softmax'))
35+
model.add(Dense(num_classes))
4136
model.compile(loss=config.loss, optimizer=config.optimizer,
4237
metrics=['accuracy'])
4338

4439
# Fit the model
4540
model.fit(X_train, y_train, epochs=config.epochs, validation_data=(X_test, y_test),
46-
callbacks=[WandbCallback(data_type="image", labels=signdata.letters)])
41+
callbacks=[WandbCallback(data_type="image", labels=signdata.letters)])

0 commit comments

Comments
 (0)