def calculateResult(gamma, seed, gradual=True): np.random.seed(seed) startTime = time.time() spntf = SP_NTF.SP_NTF(X, R=R, alpha=alpha, maxinner=INNER_ITER) # instance of SP_NTF class Yinfo = spntf.computeDecomp(gamma=gamma, gradual=gradual) totalTime = time.time() - startTime return {"gamma": gamma, "nnz": tensorTools.countTensorNNZ(spntf.M[0]), "compTime": totalTime, "iterInfo": Yinfo, "fms": TM.greedy_fms(spntf.M[0])}
def calculateValues(TM, M): fms = TM.greedy_fms(M) fos = TM.greedy_fos(M) nnz = tensorTools.countTensorNNZ(M) return fms, fos, nnz