+- Python Forum (https://python-forum.io)
+-- Forum: Python Coding (https://python-forum.io/forum-7.html)
+--- Forum: Data Science (https://python-forum.io/forum-44.html)
+--- Thread: sklearn regression to excel (/thread-17497.html)
sklearn regression to excel - punksnotdead - Apr-13-2019
#I'm using the following to generate linear model coefficients
#I'm only showing some of the code here
#They print out OK in the DOS window
import numpy as np
from sklearn.preprocessing import PolynomialFeatures
from sklearn import linear_model
clf = linear_model.LinearRegression()
print("Coefficients: ", clf.coef_)#But how can I write the coefficients out to Excel using xlsxwriter...import xlsxwriter
workbook = xlsxwriter.Workbook('Test.xlsx')
worksheet = workbook.add_worksheet('MySheet')
#code goes here
workbook.close()#I'm not sure how to do this and been trying for hours# Trying worksheet.write (row, col, clf.coef_) just throws an error
# Think it might have something to do with it being an array.
I'm a real newbie here so hopelessly lost.
Any help greatly appreciated even if just a suggestion. Thanks
RE: sklearn regression to excel - punksnotdead - Apr-14-2019
I figured it out, can just refer to clf.coef_ array elements like this
worksheet.write (row, col, clf.coef[0])
worksheet.write (row, col, clf.coef[1])
worksheet.write (row, col, clf.coef[2])
...