Advertisement
Guest User

Untitled

a guest
Apr 21st, 2018
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.15 KB | None | 0 0
  1. def find_txt_files(directory, lowercase, dictionary_stat):
  2.     # Поиск .txt файлов в директории
  3.  
  4.     # directory - директория с файлами для обучения
  5.     # lowercase - переменная, отвечающая за приведение слов, сохраняемых
  6.     # в модели, к нижнему регистру
  7.     # dictionary_stat - словарь с моделью
  8.     # ---------------------------------------------------------------- #
  9.  
  10.     # Поиск файлов в директории
  11.     for top, dirs, files in os.walk(directory):
  12.         for i in files:
  13.             # Путь к файлу
  14.             path = str(os.path.join(top, i))
  15.  
  16.             # Разделяем имя файла на 2 части: заголовок и формат
  17.             filename, file_extension = os.path.splitext(path)
  18.             # Если .тхт, то обрабатываем его
  19.             if file_extension == '.txt':
  20.                 # Обрабатываемый файл
  21.                 file = str(os.path.normpath(path))
  22.                 train(file, lowercase, dictionary_stat)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement