TencentAILabHealthcare / spatialID

34 stars 5 forks source link

如何保存DNN_model训练完毕之后的marker_genes? #12

Open UnseenAnchor opened 6 months ago

UnseenAnchor commented 6 months ago

在利用scRNA训练完dnn model之后,需要保存checkpoint,我看您保存了model optimizer epoch train_loss valid_loss valid_metric lr best_val marker_genes label_names ,这其中marker_genes 是如何筛选出来保存的呢?

SilversH commented 5 months ago

你好,感谢对我们工作的关注。这部分可以使用scRNA数据的全部基因(或和空间组的交集部分),如果有例如需要使用部分marker的先验也可以取对应部分的marker。根据我们的测试,在大部分情况下使用scRNA数据的全部基因已经能获得很好的效果。