Update demo_rom_manager_nn.py

KratosMultiphysics · Jan 30, 2024 · e8a20a0 · e8a20a0
1 parent a1ec10d
commit e8a20a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py b/rom_application/RomManager_cantilever_NN/demo_rom_manager_nn.py
@@ -85,7 +85,7 @@ def GetRomManagerParameters():
     general_rom_manager_parameters = KratosMultiphysics.Parameters("""{
             "rom_stages_to_train" : ["ROM"],             // ["ROM","HROM"]
             "rom_stages_to_test" : [],              // ["ROM","HROM"]
-            "rom_stages_nn_fit"  : ["GenValData","GenTrainData","TrainNN"],      // ["GenValData","GenTrainData","TrainNN"]
+	    "type_of_decoder" : "ann_enhanced",           // "linear" "ann_enhanced",  TODO: add "quadratic"
             "paralellism" : null,                        // null, TODO: add "compss"
             "projection_strategy": "galerkin",            // "lspg", "galerkin", "petrov_galerkin"
             "assembling_strategy": "global",            // "global", "elemental"