typo

AWehrhahn · May 13, 2021 · 26cfdb4 · 26cfdb4
1 parent 470e05b
commit 26cfdb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/paper/continuum_neural_network.py b/examples/paper/continuum_neural_network.py
@@ -238,7 +238,7 @@ def load(self, filename):
     # clf = PLSRegression(max_degree + 1)
     # clf.fit(XS_train, y_train)
     clf = ContinuumModel(verbose=True, loss="mse")
-    history = clf.fit(XS_train, y_train, validation_data=(X_validate, y_validate))
+    history = clf.fit(X_train, y_train, validation_data=(X_validate, y_validate))
 
     clf.model.save("model.dat")