Bug in GAMNET about pyhealth HOT 1 CLOSED

Hi @bbb801

Could you please modify the dynamic memory into:

        # dynamic memory
        DM_keys = queries[:, :-1, :]
        DM_values = prev_drugs[:, :-1, :]

which would make the code runs through. These two lines mean that we use the drugs from visit 1 to visit N-1, so the DM_keys and DM_values should both have :-1.

The above is for the GAMENetLayer class. Also, we have improved the GAMENet class, which replaces the target calculation blocks:

# previous block
curr_drugs = [p[-1] for p in drugs_hist]
curr_drugs = batch_to_multihot(curr_drugs, label_size)
curr_drugs = curr_drugs.to(self.device)

# new block
curr_drugs = self.prepare_labels(drugs, self.label_tokenizer)

Please refer to our current new master branch.

from pyhealth.