Ejemplo n.º 1
0
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())