我在 Tensorflow 中使用 GMM 时遇到了这个奇怪的问题。
InvalidArgumentError (see above for traceback): Cholesky decomposition was not successful. The input might not be valid.
我使用的代码如下所示。
from tensorflow.contrib.factorization.python.ops import gmm as gmm_lib
import random
import numpy as np
import tensorflow as tf
class DataGen:
start = 0
batch_size = 100
end = start + batch_size
def __init__(self, points):
self.points = points
def next_batch(self):
start = self.start
end = self.end
if end <= self.points.shape[0]:
self.start += self.batch_size
self.end += self.batch_size
return tf.constant(self.points[start: end]), None
mu, sigma = 0, 0.1
x_1d = 10*np.random.randn(500, 1).astype('f') + 50
data_gen = DataGen(x_1d)
gmm = gmm_lib.GMM(1,random_seed=0)
gmm.fit(input_fn=data_gen.next_batch)
有解决这个问题的想法吗？谢谢。