From 9c1e77d0c8e323930e0d26b7fdf522dbe934b932 Mon Sep 17 00:00:00 2001 From: niektuytel <46424968+niektuytel@users.noreply.github.com> Date: Tue, 23 Mar 2021 08:50:31 +0100 Subject: [PATCH] clean imports There where some unused imports, and set `np.cov(X, rowvar=False)` in place of `calculate_covariance_matrix(X)` to make the file more simple. --- .../unsupervised_learning/gaussian_mixture_model.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/mlfromscratch/unsupervised_learning/gaussian_mixture_model.py b/mlfromscratch/unsupervised_learning/gaussian_mixture_model.py index 17f88722..eae8a699 100644 --- a/mlfromscratch/unsupervised_learning/gaussian_mixture_model.py +++ b/mlfromscratch/unsupervised_learning/gaussian_mixture_model.py @@ -1,11 +1,6 @@ -from __future__ import division, print_function +import numpy as np import math from sklearn import datasets -import numpy as np - -from mlfromscratch.utils import normalize, euclidean_distance, calculate_covariance_matrix -from mlfromscratch.utils import Plot - class GaussianMixtureModel(): """A probabilistic clustering method for determining groupings among data samples. @@ -37,7 +32,7 @@ def _init_random_gaussians(self, X): for i in range(self.k): params = {} params["mean"] = X[np.random.choice(range(n_samples))] - params["cov"] = calculate_covariance_matrix(X) + params["cov"] = np.cov(X, rowvar=False) self.parameters.append(params) def multivariate_gaussian(self, X, params):