Huhujingjing
/

custom-mxm

Graph Machine Learning

feature-extraction

Model card Files Files and versions

Huhujingjing commited on Jul 2, 2023

Commit

263ef5a

·

1 Parent(s): 3e2d1c7

Upload model

Files changed (1) hide show

modeling_mxm.py +3 -3

modeling_mxm.py CHANGED Viewed

@@ -1015,10 +1015,10 @@ class MXMModel(PreTrainedModel):
         self.output +=('-' * 40 + '\n'+'predicted result: \n'+'{}\n'.format(pred_df))
         self.output +=('-' * 40)
-        pred_df.to_csv(os.path.join(result_dir, 'gcn.csv'), index=False)
-        self.output +=('\nsave predicted result to {}\n'.format(os.path.join(result_dir, 'gcn.csv')))
-        return self.output
 if __name__ == "__main__":

         self.output +=('-' * 40 + '\n'+'predicted result: \n'+'{}\n'.format(pred_df))
         self.output +=('-' * 40)
+        # pred_df.to_csv(os.path.join(result_dir, 'gcn.csv'), index=False)
+        # self.output +=('\nsave predicted result to {}\n'.format(os.path.join(result_dir, 'gcn.csv')))
+        return self.output, pred_df
 if __name__ == "__main__":