complex_shift_autoencoder.py [385:404]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def plot_multiple_transformations(self, param_name='angle', indices=None, train_set=False, save_name=None):
        """Plots all rotated reconstructions for given samples"""
        if indices is None:
            n_samples = min(len(self.data.X_orig_train), len(self.data.X_orig_test))
            indices = np.random.randint(low=0, high=n_samples, size=5)
        X = (
            self.data.X_orig_train[indices]
            if train_set
            else self.data.X_orig_test[indices]
        ).float()
        title = (
            "Translations" if param_name=='angle' != "angle" else "Rotations"
        )

        plot.plot_transformations_complex(
            X,
            self,
            title,
            save_name=save_name,
            param_name=param_name,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



weakly_complex_shift_autoencoder.py [204:223]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def plot_multiple_transformations(self, param_name='angle', indices=None, train_set=False, save_name=None):
        """Plots all rotated reconstructions for given samples"""
        if indices is None:
            n_samples = min(len(self.data.X_orig_train), len(self.data.X_orig_test))
            indices = np.random.randint(low=0, high=n_samples, size=5)
        X = (
            self.data.X_orig_train[indices]
            if train_set
            else self.data.X_orig_test[indices]
        ).float()
        title = (
            "Translations" if param_name=='angle' != "angle" else "Rotations"
        )

        plot.plot_transformations_complex(
            X,
            self,
            title,
            save_name=save_name,
            param_name=param_name,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -