c753a3bf6c768eb45c0b4bad556161b7389fde65,finetune/util/imbalance.py,,class_weight_tensor,#Any#Any#Any#,45

Before Change

    
    class_weight_arr = np.ones(target_dim, dtype=np.float32)
    for class_name, class_weight in class_weights.items():
        idx = LabelEncoder.transform(label_encoder, [class_name])[0]
        class_weight_arr[idx] = class_weight
    class_weight_tensor = tf.convert_to_tensor(class_weight_arr)
    return class_weight_tensor

After Change

    Convert from dictionary of class weights to tf tensor
    
    class_weight_arr = np.ones(target_dim, dtype=np.float32)
    for i, cls in enumerate(label_encoder.target_labels):
        class_weight_arr[i] = class_weights.get(cls, 1.0)

    class_weight_tensor = tf.convert_to_tensor(class_weight_arr)
    return class_weight_tensor

Instances