"buildA_true" in "train_single_step.py"

nnzhan / MTGNN

MIT License

762 stars 216 forks source link

"buildA_true" in "train_single_step.py" #6

Closed sleeperzjf closed 3 years ago

sleeperzjf commented 3 years ago

If you are going to provide a choice over buildA_true, you might as well add the type-changing codes to train_single_step.py as you did in train_multi_step.py:

nnzhan commented 3 years ago

thanks, we will improve it.