Multivariante/de Regresión Lineal Múltiple en Scikit Aprender?

Tengo un conjunto de datos (dataTrain.csv & dataTest.csv).archivo csv con este formato:

Temperature(K),Pressure(ATM),CompressibilityFactor(Z)
273.1,24.675,0.806677258
313.1,24.675,0.888394713
...,...,...

Y capaces de construir un modelo de regresión y predicción con este código:

import pandas as pd
from sklearn import linear_model

dataTrain = pd.read_csv("dataTrain.csv")
dataTest = pd.read_csv("dataTest.csv")
# print df.head()

x_train = dataTrain['Temperature(K)'].reshape(-1,1)
y_train = dataTrain['CompressibilityFactor(Z)']

x_test = dataTest['Temperature(K)'].reshape(-1,1)
y_test = dataTest['CompressibilityFactor(Z)']

ols = linear_model.LinearRegression()
model = ols.fit(x_train, y_train)

print model.predict(x_test)[0:5]

Sin embargo, lo que quiero hacer es de regresión multivariante. Así, el modelo será CompressibilityFactor(Z) = intercept + coef*Temperature(K) + coef*Pressure(ATM)

Cómo hacer que en scikit-learn?

Sólo tienes que incluir tanto la Temperatura y la Presión en su xtrain, xtest. x_train = dataTrain[["Temperature(K)", "Pressure(ATM)"]] y después de la misma para x_test.

OriginalEl autor Drizzer Silverberg | 2017-02-05

1 Kommentar

  1. 17

    Si el código anterior funciona para univariante, pruebe este

    import pandas as pd
    from sklearn import linear_model
    
    dataTrain = pd.read_csv("dataTrain.csv")
    dataTest = pd.read_csv("dataTest.csv")
    # print df.head()
    
    x_train = dataTrain[['Temperature(K)', 'Pressure(ATM)']].reshape(-1,2)
    y_train = dataTrain['CompressibilityFactor(Z)']
    
    x_test = dataTest[['Temperature(K)', 'Pressure(ATM)']].reshape(-1,2)
    y_test = dataTest['CompressibilityFactor(Z)']
    
    ols = linear_model.LinearRegression()
    model = ols.fit(x_train, y_train)
    
    print model.predict(x_test)[0:5]
    DataFrames no tiene un reshape función. Para ejecutar el código anterior tengo que usar values primera, por ejemplo, la x_train = dataTrain[['Temperature(K)', 'Pressure(ATM)']].values.reshape(-1,2).

    OriginalEl autor piRSquared

Kommentieren Sie den Artikel

Bitte geben Sie Ihren Kommentar ein!
Bitte geben Sie hier Ihren Namen ein

Pruebas en línea