Saddle point in work

Author: niravshah241

demo/saddle_point_dlrbnicsx/dlrbnicsx_saddle_point.py

@@ -1,5 +1,4 @@
 import time
-import abc
 
 import dolfinx
 import ufl
@@ -12,13 +11,22 @@
 
 from mpi4py import MPI
 from petsc4py import PETSc
+from smt.sampling_methods import LHS
 
 import numpy as np
 import itertools
 import abc
 import matplotlib.pyplot as plt
 import os
 
+from dlrbnicsx.neural_network.neural_network import HiddenLayersNet
+from dlrbnicsx.activation_function.activation_function_factory import Tanh, Sigmoid
+from dlrbnicsx.dataset.custom_dataset import CustomDataset
+from dlrbnicsx.interface.wrappers import DataLoader, save_model, load_model, \
+    save_checkpoint, load_checkpoint, get_optimiser, get_loss_func
+from dlrbnicsx.train_validate_test.train_validate_test import \
+    train_nn, validate_nn, online_nn, error_analysis
+
 class ParametricProblem(abc.ABC):
     # Define FEM problem
     def __init__(self, mesh):
@@ -720,3 +728,4 @@ def generate_ann_output_set(problem, reduced_problem, input_set, mode=None):
 elapsed_time = end_time - start_time
 os.system(f"rm {checkpoint_path_u}")
 print(f"Training time (u): {elapsed_time}")
+