reset for bloomberg

Author: lukas

examples/keras-perceptron/nn.py

@@ -0,0 +1,29 @@
+import tensorflow as tf
+import wandb
+
+# logging code
+run = wandb.init()
+config = run.config
+
+# load data
+(X_train, y_train), (X_test, y_test) = tf.keras.datasets.mnist.load_data()
+img_width = X_train.shape[1]
+img_height = X_train.shape[2]
+
+# one hot encode outputs
+y_train = tf.keras.utils.to_categorical(y_train)
+y_test = tf.keras.utils.to_categorical(y_test)
+labels = [str(i) for i in range(10)]
+
+num_classes = y_train.shape[1]
+
+# create model
+model = tf.keras.models.Sequential()
+model.add(tf.keras.layers.Flatten(input_shape=(img_width, img_height)))
+model.add(tf.keras.layers.Dense(num_classes))
+model.compile(loss='mse', optimizer='adam',
+              metrics=['accuracy'])
+
+# Fit the model
+model.fit(X_train, y_train, epochs=10, validation_data=(X_test, y_test),
+          callbacks=[wandb.keras.WandbCallback(data_type="image", labels=labels, save_model=False)])

examples/keras-perceptron/wandb/settings

@@ -1,4 +1,4 @@
 [default]
-entity: qualcomm
-project: digits-sep9
+entity: bloomberg-class
+project: digits
 base_url: https://api.wandb.ai