docs: public float[][] getTheta() {
          float[][] theta = new float[documents.length][K];
  
          if (SAMPLE_LAG > 0) {
              for (int m = 0; m < documents.length; m++) {
                  for (int k = 0; k < K; k++) {
                      theta[m][k] = thetasum[m][k] / numstats;
                  }
              }
  
          } else {
              for (int m = 0; m < documents.length; m++) {
                  for (int k = 0; k < K; k++) {
                      theta[m][k] = (nd[m][k] + alpha) / (ndsum[m] + K * alpha);
                  }
              }
          }
  
          return theta;
      }