SIMEXP / fmri_predict

predicting fmri activaties from connectome
13 stars 11 forks source link

predict resting-state using saved model #15

Open zhangyu2ustc opened 5 years ago

zhangyu2ustc commented 5 years ago

as one type of transfer learning, we will save the best model from ALL tasks, and predict the state transition in resting-state data

zhangyu2ustc commented 5 years ago

First, retrain the GCN model and save the best one:

''' from . import checkmat as checkmate best_ckpt_saver = checkmate.BestCheckpointSaver(save_dir=path,num_to_keep=3,maximize=True,saver=self.op_saver) best_ckpt_saver.handle(accuracy, sess, self.global_step) '''

Problem: it deletes some of the models because of conflicting of existing saver: ''' self.op_saver.save(sess, path, global_step=self.global_step) '''

zhangyu2ustc commented 5 years ago

the most common states from resting-state prediction are: motor, language and social

zhangyu2ustc commented 5 years ago

guide to restore any pre-trained model and use it for prediction, fine-tuning or further training: https://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-complete-tutorial/