3940f2aab97e571e99177d5b61700094637301f5,Generator/gan_main.py,GAN,main,#GAN#,151

Before Change

        // Generate injection codes.
        lst_scripts = self.train(df_genes, df_sigs)
        self.util.print_message(NOTE, "Generated injection codes: {}".format(lst_scripts))

After Change

        df_sigs = pd.read_csv(sig_path, encoding="utf-8").fillna("")
        df_selected_sigs = df_sigs[(df_sigs[0] == eval_place)]

        if len(df_selected_sigs) != 0:
            df_genes = pd.read_csv(self.genes_path, encoding="utf-8").fillna("")

            // Generate injection codes.
            lst_scripts = self.train(df_genes, df_sigs)
            self.util.print_message(NOTE, "Generated injection codes: {}".format(lst_scripts))
        else:
            self.util.print_message(WARNING, "Signature of {} do not include in {}.".format(eval_place, sig_path))
            self.util.print_message(WARNING, "Skip process of Generative Adversarial Networks")

Instances