¿Cómo puedo resolver un IndexError en el ciclo train_y?
No entiendo dónde está mi error:
“`
train_dataset = TensorDataset(input_ids, attention_masks, labels)
print(‘target train 0:’, len(np.where(train_y == 0)[0]))
print(‘target train 1:’, len(np.where(train_y == 1)[0]))
print(‘target train 2:’, len(np.where(train_y == 2)[0]))
print(‘target train 3:’, len(np.where(train_y == 3)[0]))
print(‘target train 4:’, len(np.where(train_y == 4)[0]))
print(‘target train 5:’, len(np.where(train_y == 5)[0]))
print(‘target train 6:’, len(np.where(train_y == 6)[0]))
target train 0: 6834
target train 1: 1200
target train 2: 0
target train 3: 4397
target train 4: 1112
target train 5: 0
target train 6: 3281
”’¿Cuántos ejemplos tiene cada clase?”’
class_sample_count = np.array(
[len(np.where(train_y == t)[0]) for t in np.unique(train_y)])
print(“¿Cuántos ejemplos tiene cada clase?\n”, class_sample_count)
>> [6834 1200 4397 1112 3281]
weight = 1. / class_sample_count
print(“pesos: “, weight)
>> pesos: [0.00014633 0.00083333 0.00022743 0.00089928 0.00030479]
samples_weight = np.array([weight[t] for t in train_y])
davy.ai
El error que estás recibiendo es un IndexError: el índice 6 está fuera de los límites para el eje 0 con tamaño 5. Este error ocurre porque se está utilizando el valor de índice 6 para la matriz de numpy ‘class_sample_count’, pero el tamaño de la matriz es solo de 5.
Para solucionar este error, debes verificar el rango de valores en ‘train_y’ para asegurarte de que esté dentro del rango de la matriz ‘class_sample_count’. También es posible que haya un error en la forma en que los datos se cargan o se procesan, por lo que podría ser una buena idea revisar todo el código y el flujo de datos en busca de otros errores.