Bucle por lotes sobre un arreglo de Python para crear un nuevo arreglo sin sobrescribir.
Quiero iterar sobre una matriz tridimensional (sequences) con forma (1134500, 1, 50)
Para hacer esto, utilizo el siguiente bucle for, que funciona bien excepto por sobrescribir los resultados del lote anterior:
batchsize = 500
for i in range(0, sequences.shape[0], batchsize):
batch = sequences[i:i+batchsize]
relevancias = lrp_model.lrp(batch)
Como resultado, quiero una matriz (relevancias) con forma (1134500, 1, 50), pero obtengo una con forma (500, 1, 50)
¿Alguien puede decirme qué está saliendo mal?
davy.ai
El problema en el código proporcionado es que el arreglo “relevances” es sobrescrito en cada iteración del ciclo. Por lo tanto, solo almacena los resultados del último lote procesado, lo que resulta en una salida de forma (500, 1, 50).
Para solucionar esto, puedes crear un arreglo vacío del tamaño deseado antes del ciclo y agregar los resultados de cada lote utilizando el índice de la iteración. Aquí hay un ejemplo:
Este código debería darte la salida deseada de un arreglo con forma (1134500, 1, 50), con los resultados de cada lote agregados al arreglo “relevances”.