Блог

Главная - Вопросы по программированию - Запуск EditNTS в Google Collab … Ошибка значения: неподдерживаемый протокол pickle: 5

Запуск EditNTS в Google Collab … Ошибка значения: неподдерживаемый протокол pickle: 5

#python #machine-learning #google-colaboratory #simplification

Вопрос:

Я пытаюсь запустить алгоритм упрощения текста под названием EditNTS. Я успешно запустил его локально на своем компьютере, удалив метод cuda(). Однако я пытаюсь запустить его в Google Collab, чтобы я мог воспользоваться преимуществами графического процессора. Однако я получаю сообщение об ошибке:

 File "/content/drive/My Drive/EditNTS-master/data.py", line 193, in __init__
    self.df = pd.read_pickle(data_path)
  File "/usr/local/lib/python3.7/dist-packages/pandas/io/pickle.py", line 182, in read_pickle
    return pickle.load(f)
ValueError: unsupported pickle protocol: 5

Я попробовал !pip install pickle5 и импортировал pickle5 as pickle внутри data.py но я все равно получаю ту же ошибку.

Кто-нибудь знает, как это исправить? Вот ссылка на Google Диск, все, что вам нужно сделать, это загрузить мои изменения в коде и запустить его в Google collab вот так.

Google Collab

P.S main.py возвращает модель editNTS, но я не могу найти функцию в коде, которая позволила бы мне использовать модель editNTS для прогнозирования каждого простого предложения из сложного документа. Если вы можете помочь мне с этим, это было бы очень полезно. Заранее спасибо, ребята.

Вот ссылка на Google Диск: https://drive.google.com/drive/folders/1R3ZpC_P-k9NOruiKhwsTRi_nlTG5yfjw?usp=sharing

Для локального: https://drive.google.com/drive/folders/1Wmmkb60fzui1YZUJpHXFbdXsJGRBPQXB?usp=sharing

Метки: