代码在下面,假如要训练20种图片,其中的19种每种都有200张图片,只有一种图片只有50张,这时候需要调用class_weight函数吗?
训练数据时达到什么比例时该调用,比例差异不显著时调用的话会不会减低训练准确率?
def calc_class_weight(total_y):
my_class_weight = class_weight.compute_class_weight("balanced", np.unique(total_y), total_y)
return my_class_weight
my_class_weight = calc_class_weight(y_train0)
model.fit(X_train, y_train, validation_split = 0.2,class_weight = my_class_weight,shuffle = True,epochs=100, batch_size=32)