Читаем Шолле стр. 81 – 119 и реализуем то, что описывается:
Используя библиотеку Keras на Python:
x in ['зеленый','салатовый']), или что-то еще для другого типа данных, и создать соответствующий столбец с бинарными значениями (Истина, если условие выполняется, и Ложь - в противном случае).
#...
model.compile(loss='mse', optimizer='rmsprop', metrics=['mae'])
model.fit(X_train, y_train, epochs=param["epochs"], batch_size=param["batch_size"])
_, accuracy = model.evaluate(X_train, y_train)
_, accuracy2 = model.evaluate(model.predict(X_test), y_test)
predictions = model.predict(X_train)
plt.figure(figsize=(10., 10.))
title = str(list((str(dense[j]) +' '+ str(activation[j]) for j in range(len(param["layers"])))))
plt.title(title + "\nepochs="+str(param["epochs"])+" batch_size="+str(param["batch_size"])+"\nточность обучения %.2f, точность на тесте %.2f" % (accuracy*100, accuracy2*100))
for i in range(len(X_train)):
plt.scatter((X_train)[i], predictions[i], c='red', label=('prediction' if i==0 else None))
plt.plot(x, y, c='blue', label=param["func_label"], marker='X')
plt.scatter(X_test, model.predict(X_test), c='green', label='тест')
plt.legend()