Commit d8eb5517 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

The data can be GMMStats not just numpy arrays

parent 2ea663bf
Pipeline #29922 passed with stage
in 12 minutes and 31 seconds
......@@ -88,8 +88,7 @@ def train(trainer, machine, data, max_iterations=50, convergence_threshold=None,
pool : :py:class:`multiprocessing.Pool` or ``None``
If given, the provided process pool will be used to parallelize the M-step of the EM algorithm
"""
data = numpy.asarray(data)
if check_inputs:
if check_inputs and isinstance(data, numpy.ndarray):
sum_data = numpy.sum(data)
if numpy.isinf(sum_data):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment