Multi-head Model

[shelllbw]

Implementation of multi-head models

• Support for multiple training sets
• multi-head training and prediction processes for mlp and cnn: mh_nn_learn , mh_nn_predict
• multi-head training and prediction processes for ae_mlp and ae_cnn: mh_ae_learn, mh_ae_predict
• multi-head training and prediction processes for gnn: mh_gnn_learn, mh_gnn_predict
• multi-head mlp: mh_mlp
• multi-head ae_mlp: mh_ae_mlp
• multi-head cnn: mh_cnn
• multi-head ae_cnn: mh_ae_cnn
• multi-head gnn: mh_gnn