Skip to content
Snippets Groups Projects
Commit d8eb5517 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

The data can be GMMStats not just numpy arrays

parent 2ea663bf
Branches
Tags
1 merge request!16Trial to implement EM with multiprocessing
Pipeline #29922 passed
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment