# GeneralGuess.py

# -*- coding: utf-8 -*-
# GeneralGuess.py

# Note: Sorry if the data in French is incorrect whereas it was provided by Google Translator.

import re

WORDS = re.compile(r'\b\w+\b')
all_words = {}

class GeneralGuess(object):
    def __init__(self, data):
        if len(data[-1]) > 1:
            self.train(data)
        else:
            self.score(data)

    def train(self, training_data):
        for category, message in training_data:
            try:
                all_words[category]
            except:
                all_words[category] = []
            cur_dict = all_words[category]
            words = WORDS.findall(message.lower())
            for word in words:
                if word[:4] not in all_words[category]:
                    all_words[category].append(word[:4])

    def score(self, message):
        weights = []
        for category in all_words:
            p = []
            for word in WORDS.findall(message.lower()):
                p.append(1 if word[:4] in all_words[category] else 0)

            try:
                weights.append([(1.0 / len(p)) * sum(p), category])
            except OverflowError:
                weights.append([0.0, category])
        weights.sort()
        print ['%s = %s' % (category, p) for p, category in weights[::-1]]
0
training_data = '''
english: Maybe we both can make a whole lot of money from home with this new idea.
english: That was a very huge house party and thanks again for inviting us.
english: Such the talk went on for over two hours, but we really needed it.
english: They look so much happier right now just to finally be living together.
english: All those years... she never expected he would be the perfect type for her.
english: At what time tomorrow should my friend show up to fix your computer?
french: Peut-être pouvons-nous tous les deux gagner beaucoup d’argent avec cette nouvelle idée.
french: C'était une très grande fête à la maison et merci encore de nous avoir invités.
french: La conversation a duré plus de deux heures, mais nous en avions vraiment besoin.
french: Ils ont l'air tellement plus heureux en ce moment, juste pour enfin vivre ensemble.
french: Toutes ces années... elle ne s'attendait pas à ce qu'il soit le type parfait pour elle.
french: À quelle heure demain mon ami devrait-il se présenter pour réparer votre ordinateur?
'''.strip().splitlines()

training_data = [data.split(': ') for data in training_data]

GeneralGuess(training_data)

print all_words

GeneralGuess("Given the very small amount of typical data for training on, this minimalistic algorithm will make an attempt to predict the category set by percentages whereas the application regarding this demo will not recognize a while lot of the words shown in this sample, but should still be impressive enough to take into consideration. Most importantly, I believe this simple project has the potential to handle far more complex tasks such as facial recognition with excellent accuracy and even generate very random photo realistic images, whereas another side feature is the removal of their counterpart similarities.")

GeneralGuess("Compte tenu de la très petite quantité de données typiques pour la formation, cet algorithme minimaliste tentera de prédire la catégorie en pourcentage alors que l’application concernant cette démo ne reconnaîtra pas beaucoup de mots affichés dans cet échantillon, mais devrait tout de même être assez impressionnant pour prendre en considération. Plus important encore, je pense que ce projet simple a le potentiel de traiter des tâches beaucoup plus complexes, telles que la reconnaissance faciale, avec une excellente précision, et même de générer des images photoréalistes très aléatoires, tandis qu'un autre aspect parallèle est la suppression des similitudes entre leurs contreparties.")