Ошибка декомпозиции Холецкого при использовании GMM в Tensorflow

Question

Я сталкиваюсь с этой странной проблемой при использовании GMM в Tensorflow.

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)

Есть идеи для решения этой проблемы? Thx.

Answer 1

Даже я сталкиваюсь с той же проблемой, но это не тот случай, когда мы используем Google Colab Laboratory! Вы можете выполнить все свои эксперименты в облаке в рамках настройки среды: [Colab Laboratory] [1]

https://colab.research.google.com/notebooks/welcome.ipynb#scrollTo=-Rh3-Vt9Nev9