ift7025-projet/Code/main.py

15 lines
477 B
Python
Raw Normal View History

2019-04-29 05:10:20 +00:00
# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
import sys
import load_datasets
import NeuralNet # importer la classe du Réseau de Neurones
import DecisionTree # importer la classe de l'Arbre de Décision
# importer d'autres fichiers et classes si vous en avez développés
# importer d'autres bibliothèques au besoin, sauf celles qui font du machine learning
dt = DecisionTree.DecisionTree()
dt.train(train,train_labels)
dt.tree