Mãos à Obra: Aprendizado de Máquina com Scikit-Learn e TensorFlow

114 | Capítulo 4: Treinando Modelos

O código equivalente utilizando o Scikit-Learn se parece com isso:

3

>>> from sklearn.linear_model import LinearRegression

>>> lin_reg = LinearRegression()

>>> lin_reg.fit(X, y)

>>> lin_reg.intercept_, lin_reg.coef_

(array([ 4.21509616]), array([[ 2.77011339]]))

>>> lin_reg.predict(X_new)

array([[ 4.21509616],

[ 9.75532293]])

Complexidade Computacional

O Método dos Mínimos Quadrados calcula o inverso de X

T

· X, que é a matriz n × n

(em que n é o número de características). A complexidade computacional de inverter

tal matriz é tipicamente de O(n

2,4

) a O(n

3

) (dependendo da implementação). Em outras

palavras, se dob