In [1]:
import numpy as np
In [ ]:
def estimativaGaussian(X):
'''
Esta função calcula a média e a variância de todos elementos, dado um vetor.
Parâmetro (X): uma matriz de dados X.
Retorna (Mu, Sigma):
Mu é um vetor de dimensão N que contém a média de todas as características N.
Sigma é um vetor de dimensão N com as variâncias de todas as características.
'''
mu = X.mean(axis=0)
sigma = X.var(axis=0)
return mu, sigma