55 lines
2.6 KiB
Python
55 lines
2.6 KiB
Python
"""
|
|
Note: I code in english but comment in French !
|
|
"""
|
|
import data_preparation
|
|
from HMM import HMM
|
|
import utils
|
|
|
|
|
|
if __name__ == '__main__':
|
|
numeric_french_text = data_preparation.prepare_data('french.txt')
|
|
numeric_english_text = data_preparation.prepare_data('english.txt')
|
|
numeric_italian_text = data_preparation.prepare_data('italian.txt')
|
|
|
|
lambda_fr = HMM('matrice_emission.xls', numeric_french_text)
|
|
lambda_en = HMM('matrice_emission.xls', numeric_english_text)
|
|
lambda_it = HMM('matrice_emission.xls', numeric_italian_text)
|
|
|
|
numeric_french_word = data_preparation.get_text_in_alphabet_index_form('probablement')
|
|
|
|
# On prend le premier mot de la liste (pomme)
|
|
word = numeric_french_word[0]
|
|
|
|
res_fr, _ = lambda_fr.forward(word)
|
|
res_en, _ = lambda_en.forward(word)
|
|
res_it, _ = lambda_it.forward(word)
|
|
|
|
proba_fr = utils.normalize_probabilities(res_fr, res_en, res_it, res_fr)
|
|
proba_en = utils.normalize_probabilities(res_fr, res_en, res_it, res_en)
|
|
proba_it = utils.normalize_probabilities(res_fr, res_en, res_it, res_it)
|
|
|
|
print('Résultats forward ---------------------------------------------------')
|
|
print(f'FR={proba_fr}, EN={proba_en}, IT={proba_it}, Conclusion={max(proba_fr, proba_en, proba_it)}')
|
|
|
|
res_back_fr, _ = lambda_fr.backward(word)
|
|
res_back_en, _ = lambda_en.backward(word)
|
|
res_back_it, _ = lambda_it.backward(word)
|
|
|
|
proba_back_fr = utils.normalize_probabilities(res_back_fr, res_back_en, res_back_it, res_back_fr)
|
|
proba_back_en = utils.normalize_probabilities(res_back_fr, res_back_en, res_back_it, res_back_en)
|
|
proba_back_it = utils.normalize_probabilities(res_back_fr, res_back_en, res_back_it, res_back_it)
|
|
|
|
print('Résultat backward ---------------------------------------------------')
|
|
print(f'FR={proba_back_fr}, EN={proba_back_en}, IT={proba_back_it}, Conclusion={max(proba_back_fr, proba_back_en, proba_back_it)}')
|
|
|
|
words_text_1 = data_preparation.prepare_data('texte_1.txt')
|
|
words_text_2 = data_preparation.prepare_data('texte_2.txt')
|
|
words_text_3 = data_preparation.prepare_data('texte_3.txt')
|
|
|
|
text_1_result = utils.forward_detection_with_text(lambda_fr, lambda_en, lambda_it, words_text_1)
|
|
text_2_result = utils.forward_detection_with_text(lambda_fr, lambda_en, lambda_it, words_text_2)
|
|
text_3_result = utils.forward_detection_with_text(lambda_fr, lambda_en, lambda_it, words_text_3)
|
|
|
|
print('Résultat sur les textes ----------------------------------------------')
|
|
print(f'texte 1 {text_1_result}, texte 2 {text_2_result}, texte 3 {text_3_result}')
|