Académique Documents
Professionnel Documents
Culture Documents
:ml-classification
SKLEARN, le meilleur package pour faire du machine Learning avec Python, Tous les
modèles, et tous les algorithmes de machine Learning ont déjà été implémentés avec une
architecture orientée objet, chaque modèle disposant de sa propre classe. KNN,
LinearRegression, Decision Trees, Support vector machines, etc.
titanic = sns.load_dataset('titanic')
titanic.shape
titanic.head()
Etape 5 : créer la fonction survie et test le modèles sur des nouvelles
données
Maintenant on va faire la chose que vous attendez depuis le début de cours on va écrire une
fonction qui pour déterminer si oui ou non vous auriez survécu au naufrage du Titanic
On va créer une fonction survie () dans laquelle on va faire passer notre modèle ainsi qu’une
classe de passagers une valeur pour notre sex, une valeur pour notre âge
def survie(model, pclass=1, sex=0, age=24):
Dans cette fonction on va créer un tableau Numpy qui comprend ces différents éléments :
pclass, sex, âge
x= np.array([pclass, sex, age]).reshape(1,3)
Nb : il faut veiller à entrer ces données dans le bon ordre c’est-à-dire le même que celui avec
le cas de la machine a été entrainé
print(model.predict(x))
Et finalement on va appeler notre fonction pour savoir si oui ou non vous auriez survécu au
naufrage du Titanic.