UM2N package

Subpackages

• UM2N.generator package
  • Submodules
  • UM2N.generator.burgers_solver module
    • BurgersSolver
      • BurgersSolver.get_error()
      • BurgersSolver.monitor_function()
      • BurgersSolver.project_u_()
      • BurgersSolver.solve_problem()
  • UM2N.generator.equation_generator module
    • HelmholtzEqGenerator
      • HelmholtzEqGenerator.discretise()
    • PoissonEqGenerator
      • PoissonEqGenerator.discretise()
    • RandHelmholtzEqGenerator
      • RandHelmholtzEqGenerator.discretise()
    • RandPoissonEqGenerator
      • RandPoissonEqGenerator.discretise()
  • UM2N.generator.equation_solver module
    • EquationSolver
      • EquationSolver.solve_eq()
  • UM2N.generator.mesh_generator module
    • MeshGenerator
      • MeshGenerator.get_grad_norm()
      • MeshGenerator.get_grad_phi()
      • MeshGenerator.get_gradient()
      • MeshGenerator.get_hessian()
      • MeshGenerator.get_hessian_norm()
      • MeshGenerator.get_jacobian()
      • MeshGenerator.get_jacobian_det()
      • MeshGenerator.get_monitor_val()
      • MeshGenerator.get_phi()
      • MeshGenerator.monitor_func()
      • MeshGenerator.move_mesh()
  • UM2N.generator.polymesh module
    • RandPolyMesh
      • RandPolyMesh.get_boundaries()
      • RandPolyMesh.get_curve()
      • RandPolyMesh.get_line()
      • RandPolyMesh.get_mesh()
      • RandPolyMesh.get_plane()
      • RandPolyMesh.get_points()
      • RandPolyMesh.get_rand()
      • RandPolyMesh.get_rand_points()
      • RandPolyMesh.show()
  • UM2N.generator.rand_source_generator module
    • RandSourceGenerator
      • RandSourceGenerator.get_dist_params()
      • RandSourceGenerator.get_u_exact()
      • RandSourceGenerator.set_dist_params()
  • UM2N.generator.squaremesh module
    • UnstructuredSquareMesh
      • UnstructuredSquareMesh.get_boundaries()
      • UnstructuredSquareMesh.get_corner_points()
      • UnstructuredSquareMesh.get_curve()
      • UnstructuredSquareMesh.get_line()
      • UnstructuredSquareMesh.get_mesh()
      • UnstructuredSquareMesh.get_plane()
      • UnstructuredSquareMesh.get_points()
      • UnstructuredSquareMesh.load_mesh()
      • UnstructuredSquareMesh.show()
  • UM2N.generator.swirl_demo module
    • get_c()
    • get_u_0()
  • UM2N.generator.swirl_solver module
    • SwirlSolver
      • SwirlSolver.eval_problem()
      • SwirlSolver.make_all_dirs()
      • SwirlSolver.make_log_dir()
      • SwirlSolver.make_plot_data_dir()
      • SwirlSolver.make_plot_dir()
      • SwirlSolver.make_plot_more_dir()
      • SwirlSolver.monitor_function()
      • SwirlSolver.monitor_function_on_coarse_mesh()
      • SwirlSolver.project_from_prev_u_adapt()
      • SwirlSolver.project_from_prev_u_coarse()
      • SwirlSolver.project_from_prev_u_model()
      • SwirlSolver.project_u_()
      • SwirlSolver.solve_problem()
      • SwirlSolver.solve_u()
      • SwirlSolver.solve_u_fine()
    • get_c()
    • get_log_og()
    • get_u_0()
  • UM2N.generator.swirl_solver_step module
    • SwirlSolver
      • SwirlSolver.get_error()
      • SwirlSolver.get_error_currstep()
      • SwirlSolver.monitor_function()
      • SwirlSolver.monitor_function_for_merge()
      • SwirlSolver.monitor_function_grad()
      • SwirlSolver.monitor_function_on_coarse_mesh()
      • SwirlSolver.monitor_function_pure_hessian()
      • SwirlSolver.monitor_function_smoothed_grad()
      • SwirlSolver.monitor_function_smoothed_hessian()
      • SwirlSolver.plot_fine_solution()
      • SwirlSolver.plot_res()
      • SwirlSolver.project_u_()
      • SwirlSolver.solve_problem()
      • SwirlSolver.solve_u()
      • SwirlSolver.solve_u_fine()
    • get_c()
    • get_u_0()
  • Module contents
• UM2N.helper package
  • Submodules
  • UM2N.helper.config module
    • load_yaml_to_namespace()
    • save_namespace_to_yaml()
  • UM2N.helper.helper module
    • mkdir_if_not_exist()
  • UM2N.helper.plot_util module
    • plot_compare()
  • UM2N.helper.plotting module
    • plot_attentions_map()
    • plot_attentions_map_compare()
    • plot_loss()
    • plot_mesh()
    • plot_mesh_compare()
    • plot_mesh_compare_benchmark()
    • plot_multiple_mesh_compare()
    • plot_tangle()
  • Module contents
• UM2N.loader package
  • Submodules
  • UM2N.loader.cluster_utils module
    • calc_dist()
    • get_neighbors()
    • get_new_edges()
    • sampler()
  • UM2N.loader.data_transform module
    • add_edges()
    • arg_parse()
    • process_subset()
  • UM2N.loader.dataset module
    • AggreateDataset
    • MeshData
    • MeshDataset
      • MeshDataset.get_conv_feature()
      • MeshDataset.get_conv_feature_fix()
      • MeshDataset.get_mesh_feature()
      • MeshDataset.get_x_feature()
    • MeshLoader()
    • normalise()
  • Module contents
• UM2N.model package
  • Submodules
  • UM2N.model.M2N module
    • M2N
      • M2N.forward()
  • UM2N.model.M2N_T module
    • M2N_T
      • M2N_T.forward()
  • UM2N.model.M2N_atten module
    • M2NAtten
      • M2NAtten.forward()
  • UM2N.model.M2N_dynamic_drop module
    • M2N_dynamic_drop
      • M2N_dynamic_drop.forward()
  • UM2N.model.M2N_dynamic_no_drop module
    • M2N_dynamic_no_drop
      • M2N_dynamic_no_drop.forward()
  • UM2N.model.M2T module
    • M2T
      • M2T.forward()
      • M2T.get_attention_scores()
      • M2T.transformer_monitor()
  • UM2N.model.M2T_deformer module
    • M2TDeformer
      • M2TDeformer.find_boundary()
      • M2TDeformer.fix_boundary()
      • M2TDeformer.forward()
  • UM2N.model.MRN module
    • MRN
      • MRN.forward()
      • MRN.move()
    • RecurrentGATConv
      • RecurrentGATConv.find_boundary()
      • RecurrentGATConv.fix_boundary()
      • RecurrentGATConv.forward()
  • UM2N.model.MRN_GTE module
    • MRNGlobalTransformerEncoder
      • MRNGlobalTransformerEncoder.forward()
      • MRNGlobalTransformerEncoder.move()
  • UM2N.model.MRN_LTE module
    • MRNLocalTransformerEncoder
      • MRNLocalTransformerEncoder.forward()
      • MRNLocalTransformerEncoder.move()
  • UM2N.model.MRN_atten module
    • MRNAtten
      • MRNAtten.forward()
      • MRNAtten.move()
  • UM2N.model.MRN_fix module
    • MRN_fix
      • MRN_fix.forward()
      • MRN_fix.move()
  • UM2N.model.MRN_phi module
    • MRN_phi
      • MRN_phi.forward()
    • RecurrentGATConv
      • RecurrentGATConv.forward()
  • UM2N.model.MRT module
    • MRTransformer
      • MRTransformer.forward()
      • MRTransformer.get_attention_scores()
      • MRTransformer.move()
      • MRTransformer.transformer_monitor()
  • UM2N.model.MRT_PE module
  • UM2N.model.MRT_phi module
    • MRT_phi
      • MRT_phi.forward()
  • UM2N.model.deformer module
    • RecurrentGATConv
      • RecurrentGATConv.find_boundary()
      • RecurrentGATConv.fix_boundary()
      • RecurrentGATConv.forward()
  • UM2N.model.extractor module
    • GlobalFeatExtractor
      • GlobalFeatExtractor.forward()
    • LocalFeatExtractor
      • LocalFeatExtractor.forward()
      • LocalFeatExtractor.message()
  • UM2N.model.gatdeformer module
    • DeformGAT
      • DeformGAT.find_boundary()
      • DeformGAT.fix_boundary()
      • DeformGAT.forward()
      • DeformGAT.message()
      • DeformGAT.reset_parameters()
  • UM2N.model.train_util module
    • TangleCounter
      • TangleCounter.forward()
      • TangleCounter.message()
    • count_dataset_tangle()
    • count_dataset_tangle_repeat_sampling()
    • evaluate()
    • evaluate_repeat()
    • evaluate_repeat_sampling()
    • evaluate_unsupervised()
    • get_area_loss()
    • get_face_area()
    • get_inversion_diff_loss()
    • get_inversion_loss()
    • get_inversion_node_loss()
    • get_jacob_det()
    • get_sample_tangle()
    • load_model()
    • train()
    • train_unsupervised()
  • UM2N.model.transformer_model module
    • MLP_model
      • MLP_model.forward()
    • TransformerBlock
      • TransformerBlock.forward()
    • TransformerModel
      • TransformerModel.forward()
      • TransformerModel.get_attention_scores()
  • Module contents
• UM2N.processor package
  • Submodules
  • UM2N.processor.processor module
    • MeshProcessor
      • MeshProcessor.attach_feature()
      • MeshProcessor.find_bd()
      • MeshProcessor.find_edges()
      • MeshProcessor.get_conv_feat()
      • MeshProcessor.get_conv_feat_poly()
      • MeshProcessor.save_taining_data()
      • MeshProcessor.to_train_data()
  • Module contents
• UM2N.test package
  • Submodules
  • UM2N.test.bench_burgers module
    • BurgersEvaluator
      • BurgersEvaluator.eval_problem()
      • BurgersEvaluator.get_error()
      • BurgersEvaluator.make_log_dir()
      • BurgersEvaluator.make_plot_data_dir()
      • BurgersEvaluator.make_plot_dir()
      • BurgersEvaluator.make_plot_more_dir()
      • BurgersEvaluator.project_u_()
    • get_first_entry()
    • get_log_og()
  • UM2N.test.bench_helmholtz module
  • UM2N.test.bench_swirl module
    • SwirlEvaluator
      • SwirlEvaluator.eval_problem()
      • SwirlEvaluator.get_error()
      • SwirlEvaluator.make_log_dir()
      • SwirlEvaluator.make_plot_data_dir()
      • SwirlEvaluator.make_plot_dir()
      • SwirlEvaluator.make_plot_more_dir()
      • SwirlEvaluator.plot_res()
      • SwirlEvaluator.project_u_()
      • SwirlEvaluator.solve_u()
      • SwirlEvaluator.solve_u_fine()
      • SwirlEvaluator.vis_evaluate()
    • get_log_og()
  • UM2N.test.burgers_bench module
  • UM2N.test.compare_hlmhltz module
    • compare_error()
  • UM2N.test.stat module
    • write_stat()
  • UM2N.test.tangle module
    • check_dataset_tangle()
    • plot_prediction()
    • plot_sample()
  • Module contents

Module contents