import os
import sys
from datetime import datetime
from weka import Weka
from classifier import Classifier
"""
Init training for all images in a folder
"""
if __name__ == "__main__":
print("Started in", datetime.now().time())
model = 'naive-bayes'
if len(sys.argv) > 1:
model = sys.argv[1]
features = Weka('/home/eliasheinzen/aprendizado-maquina/python/images'
).extractTo(fileName='caracteristicas')
classifier = Classifier(data=features)
if model == 'naive-bayes':
classifier.NaiveBayes()
elif model == 'decision-tree':
classifier.DecisionTree()
print("Ended time", datetime.now().time())
