diff --git a/part2_advanced_config.ipynb b/part2_advanced_config.ipynb index 688568d8..0b8b31b0 100644 --- a/part2_advanced_config.ipynb +++ b/part2_advanced_config.ipynb @@ -40,7 +40,7 @@ "outputs": [], "source": [ "X_train_val = np.load('X_train_val.npy')\n", - "X_test = np.load('X_test.npy')\n", + "X_test = np.ascontiguousarray(np.load('X_test.npy'))\n", "y_train_val = np.load('y_train_val.npy')\n", "y_test = np.load('y_test.npy', allow_pickle=True)\n", "classes = np.load('classes.npy', allow_pickle=True)" @@ -294,7 +294,6 @@ " output_dir='model_1/hls4ml_prj_2',\n", " fpga_part='xcu250-figd2104-2L-e')\n", "hls_model.compile()\n", - "X_test = np.ascontiguousarray(X_test)s\n", "y_hls = hls_model.predict(X_test)\n", "print(\"Keras Accuracy: {}\".format(accuracy_score(np.argmax(y_test, axis=1), np.argmax(y_keras, axis=1))))\n", "print(\"hls4ml Accuracy: {}\".format(accuracy_score(np.argmax(y_test, axis=1), np.argmax(y_hls, axis=1))))\n",