Update prep.py
This commit is contained in:
Родитель
33f0a46ab2
Коммит
4900c570b5
|
@ -71,10 +71,6 @@ def main(args):
|
|||
# ------------ Reading Data ------------ #
|
||||
# -------------------------------------- #
|
||||
|
||||
print("mounted_path files: ")
|
||||
arr = os.listdir(args.raw_data)
|
||||
print(arr)
|
||||
|
||||
data = pd.read_csv((Path(args.raw_data)))
|
||||
data = data[NUMERIC_COLS + CAT_NOM_COLS + CAT_ORD_COLS + [TARGET_COL]]
|
||||
|
||||
|
@ -129,4 +125,4 @@ if __name__ == "__main__":
|
|||
|
||||
mlflow.end_run()
|
||||
|
||||
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче