soorena62
20 lines
532 B
Python
20 lines
532 B
Python
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)
|