Ошибка при загрузке файла .arff с помощью scikit-learn

Question

Я хотел бы использовать формат файла атрибута-отношения (.arff) с scikit-learn для задачи классификации.Код отлично работает на компьютере с Windows 10, однако, когда я пытаюсь использовать тот же код на другом компьютере с Ubuntu (18.04.1), возникает сбивающая с толку ошибка.Вот код для загрузки файла arff:

import arff, numpy as np
dataset = arff.load(open('mydataset.arff'))
mydata = np.array(dataset['data'])

И ошибка, которую я получаю, такова:

Traceback (most recent call last):
  File "/home/user/Desktop/ml_classification.py", line 14, in <module>
    mydata = np.array(dataset['data'])
TypeError: 'generator' object is not subscriptable

Что может быть причиной этой ошибки и почему она толькопроисходят на одной машине, а не на другой?

Answer 1

Я предполагаю, что вы используете старую или неподдерживаемую библиотеку для ARFF.Чтобы узнать подробности о пакете ARFF, который вы используете, попробуйте pip show arff.В моей первой попытке он показал URL для сайта с кодом Google (который сейчас не существует).Попробуйте удалить текущий пакет arff и установить его на https://pypi.org/project/liac-arff/ с pip install liac-arff.Ваш код должен работать с пакетом liac-arff.