Changed maximum training degree to 6.

ANN_Data_Generator.py

@@ -45,9 +45,9 @@ class TrainingDataGenerator:
 
         """
         self._basis_list = [OrthonormalLegendre(pol_deg)
-                            for pol_deg in range(5)]
+                            for pol_deg in range(7)]
         self._quadrature_list = [Gauss({'num_nodes': pol_deg+1})
-                                 for pol_deg in range(5)]
+                                 for pol_deg in range(7)]
         self._mesh_list = [Mesh(left_bound=left_bound, right_bound=right_bound,
                                 num_ghost_cells=0, num_cells=2**exp)
                            for exp in range(3, 12)]
@@ -227,7 +227,7 @@ class TrainingDataGenerator:
                 shift += mesh.cell_len/2
 
             # Calculate basis coefficients for stencil
-            polynomial_degree = np.random.randint(1, high=5)
+            polynomial_degree = np.random.randint(1, high=7)
             projection = do_initial_projection(
                 init_cond=init_cond, mesh=mesh,
                 basis=self._basis_list[polynomial_degree],