c = 0
bands_pseudo_label = []
angle_pseudo_label = []
icebergs_pseudo_label = []
features_pseudo_label = []
for i in prediction:
    if i[0] > 0.9995:
        bands_pseudo_label.append(bands_validate[c])
        angle_pseudo_label.append(angle_validate[c])
        features_pseudo_label.append(features_validate[c])
        icebergs_pseudo_label.append(0)
    elif i[1] > 0.9995:
        bands_pseudo_label.append(bands_validate[c])
        angle_pseudo_label.append(angle_validate[c])
        features_pseudo_label.append(features_validate[c])
        icebergs_pseudo_label.append(1)
    c+=1