Commit 9e2cc3e2 authored by dasharatha.vamshi's avatar dasharatha.vamshi

added final pipeline

parent d73b79d8
# bsj-models-for-rubbers
# To run the script use the command:
# python main.cpython-310.pyc -r "raw-file-path" -v "viscosity-file-path" -sr "skip rows for raw file" -sv "skip rows for viscosity file" -index "size index no" -m "model path"
# Eg : python main.cpython-310.pyc -r fy664g_raw.csv -v fy664g-viscosity.xlsx -sr 0 -sv 2 -index 3294 -m models\fy664g
import py_compile
py_compile.compile('main.py')
\ No newline at end of file
......@@ -1916,8 +1916,8 @@ def start_prediction(raw_path, viscosity_path, index_no, raw_skip_rows, viscosit
df_grouped = merged_all_sections(df_sheet_grouped, df_mixer_grouped, df_extruder_grouped, df_bof_grouped,
df_pickup_grouped, viscosity_df)
# load_and_predict(df_grouped, index_no, model_path)
model_trainer(df_grouped, index_no, model_path)
load_and_predict(df_grouped, index_no, model_path)
# model_trainer(df_grouped, index_no, model_path)
except Exception as er:
logger.error(f"Error in main prediction: {str(er)}")
raise Exception(str(er))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment