import pickle

# Lecture de la base de données globale.
#
# Il faudra vraisemblablement adapter le chemin vers le fichier wooldridge.pkl.

fid = open('C:/Users/claire.loupias/Desktop/wooldridge23.pkl', 'rb')
alldatasets = pickle.load(fid)
fid.close()


# Lecture de la base de données nécessaire à l'exercice.

dataset = alldatasets['sleep75']

#
# Question 1.
#

import statsmodels.formula.api as smf

results = smf.ols('sleep ~ totwrk', data=dataset).fit()

print(results.summary())


#
# Question 2.
#

beta = results.params
pred = 120*beta[1]

print('En travaillant deux heures de plus, on perd '
      'en moyenne '+f'{abs(pred):.2f}'+' minutes de sommeil.')