Не могу обучить набор данных с нейролабом - PullRequest
0 голосов
/ 26 января 2020

Я пытаюсь использовать neurolab для классификации данных, но я получил ошибку 'AssertionError', сейчас мой код

names = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k']
dataset = pandas.read_csv('./glass.data', sep= ',', names = names)
# dataset = np.genfromtxt('./glass.data', delimiter=',')
Y = np.array(dataset['k'].values)
del dataset['k']
del dataset['a']
X = np.array(dataset.values)
multilayer = neuro.net.newff([[1, 7]], [10, 1])
multilayer.trainf = neuro.train.train_gd

error = multilayer.train(X, Y, epochs = 1000, show = 100, goal = 0.01)

, где Y - класс, а X - атрибуты (я бы разбил на обучающие и тестируемые потом). мой X содержит данные:

[[ 1.52101 13.64     4.49    ...  8.75     0.       0.     ]
 [ 1.51761 13.89     3.6     ...  7.83     0.       0.     ]
 [ 1.51618 13.53     3.55    ...  7.78     0.       0.     ]
 ...
 [ 1.52065 14.36     0.      ...  8.44     1.64     0.     ]
 [ 1.51651 14.38     0.      ...  8.48     1.57     0.     ]
 [ 1.51711 14.23     0.      ...  8.62     1.67     0.     ]]

Я пробовал также с изменением формы на X и Y, и это также не работает, если кто-то хочет просмотреть весь набор данных, вот ссылка: https://archive.ics.uci.edu/ml/machine-learning-databases/glass/

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...