def read_csv_data(file_path):
df = pd.read_csv(file_path)
X = df[['倾斜角','航向角','俯仰角']].to_numpy().reshape(1,-1,3)
Y = df['ACTION'].to_numpy().reshape(1,-1)
return X,Y
def noise_au(X,Y):
X_aug, Y_aug = tsaug.AddNoise(scale=0.01).augment(X, Y)
def pool_au(X,Y):
X_aug, Y_aug = tsaug.Pool(size=2).augment(X, Y)
if __name__ == "__main__":
file_path
X,Y = read_csv_data(file_path)
noise_au(X,Y)
pool_au(X,Y)