UM2N package

Subpackages

• UM2N.generator package
  • Submodules
  • UM2N.generator.burgers_solver module
    • BurgersSolver
      • BurgersSolver.project_u_()
      • BurgersSolver.monitor_function()
      • BurgersSolver.solve_problem()
      • BurgersSolver.get_error()
  • UM2N.generator.equation_generator module
    • RandHelmholtzEqGenerator
      • RandHelmholtzEqGenerator.discretise()
    • HelmholtzEqGenerator
      • HelmholtzEqGenerator.discretise()
    • RandPoissonEqGenerator
      • RandPoissonEqGenerator.discretise()
    • PoissonEqGenerator
      • PoissonEqGenerator.discretise()
  • UM2N.generator.equation_solver module
    • EquationSolver
      • EquationSolver.solve_eq()
  • UM2N.generator.mesh_generator module
    • MeshGenerator
      • MeshGenerator.move_mesh()
      • MeshGenerator.get_grad_phi()
      • MeshGenerator.get_phi()
      • MeshGenerator.get_monitor_val()
      • MeshGenerator.get_jacobian()
      • MeshGenerator.get_jacobian_det()
      • MeshGenerator.get_gradient()
      • MeshGenerator.get_grad_norm()
      • MeshGenerator.get_hessian()
      • MeshGenerator.get_hessian_norm()
      • MeshGenerator.monitor_func()
  • UM2N.generator.rand_source_generator module
    • RandSourceGenerator
      • RandSourceGenerator.set_dist_params()
      • RandSourceGenerator.get_dist_params()
      • RandSourceGenerator.get_u_exact()
  • UM2N.generator.swirl_demo module
    • get_c()
    • get_u_0()
  • UM2N.generator.swirl_solver module
    • get_log_og()
    • get_c()
    • get_u_0()
    • SwirlSolver
      • SwirlSolver.solve_u()
      • SwirlSolver.solve_u_fine()
      • SwirlSolver.project_u_()
      • SwirlSolver.project_from_prev_u_coarse()
      • SwirlSolver.project_from_prev_u_adapt()
      • SwirlSolver.project_from_prev_u_model()
      • SwirlSolver.monitor_function()
      • SwirlSolver.monitor_function_on_coarse_mesh()
      • SwirlSolver.solve_problem()
      • SwirlSolver.eval_problem()
      • SwirlSolver.make_all_dirs()
      • SwirlSolver.make_log_dir()
      • SwirlSolver.make_plot_dir()
      • SwirlSolver.make_plot_more_dir()
      • SwirlSolver.make_plot_data_dir()
  • UM2N.generator.swirl_solver_step module
    • get_c()
    • get_u_0()
    • SwirlSolver
      • SwirlSolver.solve_u()
      • SwirlSolver.solve_u_fine()
      • SwirlSolver.project_u_()
      • SwirlSolver.monitor_function_pure_hessian()
      • SwirlSolver.monitor_function_smoothed_hessian()
      • SwirlSolver.monitor_function_grad()
      • SwirlSolver.monitor_function_smoothed_grad()
      • SwirlSolver.monitor_function_for_merge()
      • SwirlSolver.monitor_function()
      • SwirlSolver.monitor_function_on_coarse_mesh()
      • SwirlSolver.solve_problem()
      • SwirlSolver.get_error_currstep()
      • SwirlSolver.get_error()
      • SwirlSolver.plot_fine_solution()
      • SwirlSolver.plot_res()
  • UM2N.generator.unstructured_mesh module
    • UnstructuredMeshGenerator
      • UnstructuredMeshGenerator.corners
      • UnstructuredMeshGenerator.generate_mesh()
      • UnstructuredMeshGenerator.load_mesh()
    • UnstructuredSquareMeshGenerator
      • UnstructuredSquareMeshGenerator.corners
    • UnstructuredRandomPolygonalMeshGenerator
      • UnstructuredRandomPolygonalMeshGenerator.sample_uniform()
      • UnstructuredRandomPolygonalMeshGenerator.corners
  • 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_tangle()
    • plot_loss()
    • plot_mesh()
    • plot_hessian()
    • plot_attention()
    • plot_mesh_compare()
    • plot_mesh_compare_benchmark()
    • plot_multiple_mesh_compare()
    • plot_attentions_map()
    • plot_attention_on_mesh()
    • plot_attentions_map_compare()
  • Module contents
• UM2N.loader package
  • Submodules
  • UM2N.loader.cluster_utils module
    • get_neighbors()
    • calc_dist()
    • sampler()
    • get_new_edges()
    • get_neighbors_v0()
  • UM2N.loader.data_transform module
    • arg_parse()
    • add_edges()
    • process_subset()
  • UM2N.loader.dataset module
    • AggreateDataset
    • MeshDataset
      • MeshDataset.get_x_feature()
      • MeshDataset.get_mesh_feature()
      • MeshDataset.get_conv_feature()
      • MeshDataset.get_conv_feature_fix()
    • MeshData
    • MeshLoader()
    • normalise()
  • Module contents
• UM2N.model package
  • Submodules
  • UM2N.model.M2N module
    • NetGATDeform
      • NetGATDeform.forward()
    • M2N
      • M2N.forward()
  • UM2N.model.M2N_T module
    • NetGATDeform
      • NetGATDeform.forward()
    • M2N_T
      • M2N_T.forward()
  • UM2N.model.M2N_atten module
    • NetGATDeform
      • NetGATDeform.forward()
    • M2NAtten
      • M2NAtten.forward()
  • UM2N.model.M2N_dynamic_drop module
    • NetGATDeform
      • NetGATDeform.forward()
    • M2N_dynamic_drop
      • M2N_dynamic_drop.forward()
  • UM2N.model.M2N_dynamic_no_drop module
    • NetGATDeform
      • NetGATDeform.forward()
    • M2N_dynamic_no_drop
      • M2N_dynamic_no_drop.forward()
  • UM2N.model.M2T module
    • M2T
      • M2T.transformer_monitor()
      • M2T.forward()
      • M2T.get_attention_scores()
  • UM2N.model.M2T_deformer module
    • M2TDeformer
      • M2TDeformer.forward()
      • M2TDeformer.find_boundary()
      • M2TDeformer.fix_boundary()
  • UM2N.model.MRN module
    • RecurrentGATConv
      • RecurrentGATConv.forward()
      • RecurrentGATConv.find_boundary()
      • RecurrentGATConv.fix_boundary()
    • MRN
      • MRN.move()
      • MRN.forward()
  • UM2N.model.MRN_GTE module
    • MRNGlobalTransformerEncoder
      • MRNGlobalTransformerEncoder.move()
      • MRNGlobalTransformerEncoder.forward()
  • UM2N.model.MRN_LTE module
    • MRNLocalTransformerEncoder
      • MRNLocalTransformerEncoder.move()
      • MRNLocalTransformerEncoder.forward()
  • UM2N.model.MRN_atten module
    • RecurrentGATConv
      • RecurrentGATConv.forward()
      • RecurrentGATConv.find_boundary()
      • RecurrentGATConv.fix_boundary()
    • MRNAtten
      • MRNAtten.move()
      • MRNAtten.forward()
  • UM2N.model.MRN_fix module
    • RecurrentGATConv
      • RecurrentGATConv.forward()
      • RecurrentGATConv.find_boundary()
      • RecurrentGATConv.fix_boundary()
    • MRN_fix
      • MRN_fix.move()
      • MRN_fix.forward()
  • UM2N.model.MRN_phi module
    • RecurrentGATConv
      • RecurrentGATConv.forward()
    • MRN_phi
      • MRN_phi.forward()
  • UM2N.model.MRT module
    • MRTransformer
      • MRTransformer.transformer_monitor()
      • MRTransformer.move()
      • MRTransformer.forward()
      • MRTransformer.get_attention_scores()
  • UM2N.model.MRT_PE module
  • UM2N.model.MRT_phi module
    • RecurrentGATConv
      • RecurrentGATConv.forward()
    • MRT_phi
      • MRT_phi.forward()
  • UM2N.model.deformer module
    • RecurrentGATConv
      • RecurrentGATConv.forward()
      • RecurrentGATConv.find_boundary()
      • RecurrentGATConv.fix_boundary()
  • UM2N.model.extractor module
    • LocalFeatExtractor
      • LocalFeatExtractor.forward()
      • LocalFeatExtractor.message()
    • GlobalFeatExtractor
      • GlobalFeatExtractor.forward()
    • TransformerEncoder
      • TransformerEncoder.forward()
  • UM2N.model.gatdeformer module
    • DeformGAT
      • DeformGAT.reset_parameters()
      • DeformGAT.forward()
      • DeformGAT.message()
      • DeformGAT.find_boundary()
      • DeformGAT.fix_boundary()
  • UM2N.model.train_util module
    • chamfer_distance()
    • get_face_area()
    • get_inversion_loss()
    • get_inversion_diff_loss()
    • get_inversion_node_loss()
    • get_area_loss()
    • jacobLoss()
    • get_jacob_det()
    • TangleCounter
      • TangleCounter.forward()
      • TangleCounter.message()
    • print_parameter_grad()
    • train()
    • evaluate()
    • interpolate()
    • generate_samples()
    • compute_finite_difference()
    • generate_samples_structured_grid()
    • construct_graph()
    • compute_phi_hessian()
    • model_forward()
    • sample_nodes_by_monitor()
    • train_unsupervised()
    • evaluate_unsupervised()
    • get_sample_tangle()
    • count_dataset_tangle()
    • evaluate_repeat_sampling()
    • count_dataset_tangle_repeat_sampling()
    • evaluate_repeat()
    • load_model()
  • 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
    • judge_in_hull()
    • get_sample_points_in_convex_hull()
    • MeshProcessor
      • MeshProcessor.get_conv_feat_poly()
      • MeshProcessor.get_conv_feat()
      • MeshProcessor.attach_feature()
      • MeshProcessor.to_train_data()
      • MeshProcessor.find_edges()
      • MeshProcessor.find_bd()
      • MeshProcessor.save_taining_data()
  • Module contents
• UM2N.test package
  • Submodules
  • UM2N.test.bench_burgers module
    • get_log_og()
    • get_first_entry()
    • BurgersEvaluator
      • BurgersEvaluator.project_u_()
      • BurgersEvaluator.eval_problem()
      • BurgersEvaluator.get_error()
      • BurgersEvaluator.make_log_dir()
      • BurgersEvaluator.make_plot_dir()
      • BurgersEvaluator.make_plot_more_dir()
      • BurgersEvaluator.make_plot_data_dir()
  • UM2N.test.bench_swirl module
    • get_log_og()
    • SwirlEvaluator
      • SwirlEvaluator.solve_u()
      • SwirlEvaluator.solve_u_fine()
      • SwirlEvaluator.project_u_()
      • SwirlEvaluator.make_log_dir()
      • SwirlEvaluator.make_plot_dir()
      • SwirlEvaluator.make_plot_more_dir()
      • SwirlEvaluator.make_plot_data_dir()
      • SwirlEvaluator.eval_problem()
      • SwirlEvaluator.vis_evaluate()
      • SwirlEvaluator.get_error()
      • SwirlEvaluator.plot_res()
  • 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