Merge pull request #19 from XanaduAI/dev-joseph

small fixes

Author: josephbowles

src/qml_benchmarks/data/bars_and_stripes.py

@@ -23,19 +23,21 @@ def generate_bars_and_stripes(n_samples, height, width, noise_std):
         height (int): number of pixels for image height
         width (int): number of pixels for image width
         noise_std (float): standard deviation of Gaussian noise added to the pixels
+    Returns:
+        (array): data labels. -1 corresponds to a bar, +1 to a stripe.
     """
     X = np.ones([n_samples, 1, height, width]) * -1
     y = np.zeros([n_samples])
 
     for i in range(len(X)):
         if np.random.rand() > 0.5:
-            rows = np.where(np.random.rand(width) > 0.5)[0]
+            rows = np.where(np.random.rand(height) > 0.5)[0]
             X[i, 0, rows, :] = 1.0
-            y[i] = -1
+            y[i] = +1
         else:
-            columns = np.where(np.random.rand(height) > 0.5)[0]
+            columns = np.where(np.random.rand(width) > 0.5)[0]
             X[i, 0, :, columns] = 1.0
-            y[i] = +1
+            y[i] = -1
         X[i, 0] = X[i, 0] + np.random.normal(0, noise_std, size=X[i, 0].shape)
 
     return X, y

src/qml_benchmarks/models/quanvolutional_neural_network.py

@@ -77,7 +77,7 @@ def __init__(
         jit=True,
         learning_rate=0.001,
         max_steps=10000,
-        convergence_interval=10e-4,
+        convergence_interval=200,
         batch_size=32,
         random_state=42,
         scaling=1.0,