Machine Learning avec Scikit-Learn

22

Chapitre 1. Vue d’ensemble du Machine Learning

Exemple 1.1 – Entraînement et exécution d’un modèle linéaire à l’aide de Scikit-Learn

import matplotlib.pyplot as plt

import numpy as np

import pandas as pd

import sklearn.linear_model

# Chargement des données

oecd_bli = pd.read_csv("oecd_bli_2015.csv", thousands=',')

gdp_per_capita = pd.read_csv("gdp_per_capita.csv",thousands=',',delimiter='\t',

encoding='latin1', na_values="n/a")

# Préparation des données

country_stats = prepare_country_stats(oecd_bli, gdp_per_capita)

X = np.c_[country_stats["GDP per capita"]]

y = np.c_[country_stats["Life satisfaction"]]

# Visualisation des données

country_stats.plot