Prototype code of the tight-binding hamiltonian construction neural network model. Check the example_basic_method.py to construct a TB model for the InSe nanoribbon; Check the example_variation_one.py to optimize a given Wannier TB model for 2-D black phosphorus.