UofT-DSI | deep_learning - Assignment 2

01_materials/labs/lab1test.py

@@ -0,0 +1,27 @@
+import os
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+os.environ["OMP_NUM_THREADS"] = "1"
+os.environ["TF_NUM_INTRAOP_THREADS"] = "1"
+os.environ["TF_NUM_INTEROP_THREADS"] = "1"
+
+import tensorflow as tf
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Dense
+from tensorflow.keras.utils import to_categorical
+from sklearn.datasets import load_digits
+from sklearn.model_selection import train_test_split
+
+digits = load_digits()
+X = digits.data
+y = to_categorical(digits.target)
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
+
+model = Sequential([
+    Dense(64, activation="relu", input_shape=(64,)),
+    Dense(10, activation="softmax")
+])
+model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])
+
+print("Starting training…")
+model.fit(X_train, y_train, epochs=2, batch_size=32, validation_split=0.2)
+print("Done")