Fixing examples

Author: cajchristian

examples/pcovr/PCovR.py

@@ -50,7 +50,7 @@
 )
 pcovr.fit(X_scaled, y_scaled)
 T = pcovr.transform(X_scaled)
-yp = y_scaler.inverse_transform(pcovr.predict(X_scaled))
+yp = y_scaler.inverse_transform(pcovr.predict(X_scaled).reshape(-1, 1))
 
 fig, ((axT, axy), (caxT, caxy)) = plt.subplots(
     2, 2, figsize=(8, 5), gridspec_kw=dict(height_ratios=(1, 0.1))
@@ -90,7 +90,7 @@
     )
     pcovr.fit(X_scaled, y_scaled)
     T = pcovr.transform(X_scaled)
-    yp = y_scaler.inverse_transform(pcovr.predict(X_scaled))
+    yp = y_scaler.inverse_transform(pcovr.predict(X_scaled).reshape(-1, 1))
 
     axes[0, i].scatter(
         T[:, 0], T[:, 1], s=50, alpha=0.8, c=y, cmap=cmapX, edgecolor="k"
@@ -136,7 +136,7 @@
 )
 kpcovr.fit(X_scaled, y_scaled)
 T = kpcovr.transform(X_scaled)
-yp = y_scaler.inverse_transform(kpcovr.predict(X_scaled))
+yp = y_scaler.inverse_transform(kpcovr.predict(X_scaled).reshape(-1, 1))
 
 fig, ((axT, axy), (caxT, caxy)) = plt.subplots(
     2, 2, figsize=(8, 5), gridspec_kw=dict(height_ratios=(1, 0.1))

examples/pcovr/PCovR_Scaling.py

@@ -57,7 +57,7 @@
 
 pcovr_scaled = PCovR(mixing=0.5, n_components=4).fit(X_scaled, y_scaled)
 T_scaled = pcovr_scaled.transform(X_scaled)
-Yp_scaled = y_scaler.inverse_transform(pcovr_scaled.predict(X_scaled))
+Yp_scaled = y_scaler.inverse_transform(pcovr_scaled.predict(X_scaled).reshape(-1, 1))
 
 fig, ((ax1_T, ax2_T), (ax1_Y, ax2_Y)) = plt.subplots(2, 2, figsize=(8, 10))