Add All Folders
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
from pysera.preprocessing import Preprocessor
|
||||
import argparse
|
||||
import pandas as pd
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--input", required=True)
|
||||
parser.add_argument("--method", choices=["resample", "shuffle"], required=True)
|
||||
parser.add_argument("--output", required=True)
|
||||
args = parser.parse_args()
|
||||
|
||||
df = pd.read_csv(args.input)
|
||||
prep = Preprocessor(df)
|
||||
|
||||
if args.method == "resample":
|
||||
result = prep.resample()
|
||||
elif args.method == "shuffle":
|
||||
result = prep.shuffle()
|
||||
|
||||
result.to_csv(args.output, index=False)
|
||||
Reference in New Issue
Block a user