[失衡索引] #imbalance

imbalance_index.py

# Make imbalanced training samples 
print(X_train.shape)
print(Y_train.shape)

a = []
for i in range(10):
    for j in range(100):
        if i!=4:
            a.append(i*100+j)
        elif j<10:
            a.append(i*100+j)
print(a)
X_train = X_train[a,:]
Y_train = Y_train[a,:]
print(X_train.shape)
print(Y_train.shape)
# Make imbalanced training samples