def get_class_distribution(self, subject_list):
class_frequencies = np.zeros(self.n_classes)
for subj in subject_list:
labels = subj.load_labels()
mask = subj.load_ROI_mask()
class_frequencies += np.bincount(labels.flatten().astype('int'), weights=mask.flatten(),
minlength=self.n_classes)
return class_frequencies
评论列表
文章目录