Merge branch 'hls4ml-v0.5.0' into bdt

fastmachinelearning · Apr 16, 2021 · f6bd38d · f6bd38d
2 parents a23a30c + a3583d1
commit f6bd38d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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",