The PMI matrix has to be initialized for each batch because the batch's PMI matrix is combined with the current PMI matrix. This has to be added into the code.
If the matrix is not initialized then the assert statement will return an error in calc_pmi function
The PMI matrix has to be initialized for each batch because the batch's PMI matrix is combined with the current PMI matrix. This has to be added into the code.
If the matrix is not initialized then the assert statement will return an error in calc_pmi function