Turning a Pandas Dataframe to an array and evaluate Multiple Linear Regression Model

后端 未结 2 665
一向
一向 2020-12-18 07:36

I am trying to evaluate a multiple linear regression model. I have a data set like this :

\"enter

2条回答
  •  余生分开走
    2020-12-18 08:28

    y = broken_df.ground_truth.values
    X = broken_df.drop('ground_truth', axis=1).values
    X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
    linreg = LinearRegression()
    linreg.fit(X_train, y_train)
    y_pred = linreg.predict(X_test)
    print(linreg.score(X_test, y_test)
    print(classification_report(y_test, y_pred))
    

提交回复
热议问题