e8115b9d2ea60e3d7757f79fb028f778698f00fa,plasma/models/shallow_runner.py,FeatureExtractor,load_shot,#FeatureExtractor#Any#Any#,62

Before Change

            Y = []
            while(len(X) == 0):
                for i in range(length-timesteps+1):
                    if np.random.rand() < sample_prob:
                        x,y = self.get_x_y(i,shot)
                        X.append(x)
                        Y.append(y)

After Change

            np.savez(save_path,X=X,Y=Y,disr=disr)
            //print(X.shape,Y.shape)
        if sample_prob < 1.0: 
            indices = np.sort(np.random.choice(np.array(range(len(Y))),int(round(sample_prob*len(Y))),replace=False))
            X = X[indices]
            Y = Y[indices]
        return X,Y,disr

Instances