Update install requirements and readme

2020-07-30 11:19:02 +02:00
parent cd652508b9
commit 9ccbec52f7
3 changed files with 7 additions and 3 deletions
--- a/examples/glvq_iris.py
+++ b/examples/glvq_iris.py
@@ -12,7 +12,7 @@ from prototorch.modules.prototypes import Prototypes1D

 # Prepare and preprocess the data
 scaler = StandardScaler()
-x_train, y_train = load_iris(True)
+x_train, y_train = load_iris(return_X_y=True)
 x_train = x_train[:, [0, 2]]
 scaler.fit(x_train)
 x_train = scaler.transform(x_train)