!pip install neupy
from neupy import algorithms
pnn = algorithms.PNN(std=10, verbose=False)
pnn.train(X_train, y_train)
Y_train_predict = pnn.predict(X_train)
print(Y_train_predict)
r = confusion_matrix(y_train, Y_train_predict)
print(r)
fig3 = go.Figure(data=[
    go.Scatter3d(
        x=X_train[:,0], y=X_train[:,1], z=X_train[:,2], mode='markers', marker=dict(size=4, color=Y_train_predict)
        )
    ])
fig3.show()