From 62d76361e77a4afe3b7ca693fb17baed69aab002 Mon Sep 17 00:00:00 2001 From: Ilya Dziamidchyk Date: Fri, 12 Jan 2024 01:00:07 +0100 Subject: [PATCH] Added saving predicted values to a .pred file --- machine_learning/model_inference.py | 21 ++++++++++----------- 1 file changed, 10 insertions(+), 11 deletions(-) diff --git a/machine_learning/model_inference.py b/machine_learning/model_inference.py index 8d18975..ee03f16 100644 --- a/machine_learning/model_inference.py +++ b/machine_learning/model_inference.py @@ -8,20 +8,19 @@ TEST_DATA_DIR = "datasets_test" test_df_list = [] for file in os.listdir(TEST_DATA_DIR): file_path = os.path.join(TEST_DATA_DIR, file) - df = pd.read_csv(file_path, delim_whitespace=True, skiprows=1, + data_test = pd.read_csv(file_path, delim_whitespace=True, skiprows=1, names=["tbid", "tphys", "r", "vr", "vt", "ik1", "ik2", "sm1", "sm2", "a", "e", "collapsed"]) - test_df_list.append(df) + X_test = data_test.iloc[:, 1:-1].values -data_test = pd.concat(test_df_list, ignore_index=True).sample(frac=1, random_state=42) -X_test = data_test.iloc[:, 1:-1].values + imputer = SimpleImputer(strategy='mean') + X_imputed = imputer.fit_transform(X_test) -imputer = SimpleImputer(strategy='mean') -X_imputed = imputer.fit_transform(X_test) + model_filename = 'trained_model.pkl' + model = joblib.load(model_filename) -model_filename = 'trained_model.pkl' -model = joblib.load(model_filename) - -predictions = model.predict(X_test) -print(predictions) + predictions = model.predict(X_test) + data_test['prediction'] = predictions + save_path = os.path.join(TEST_DATA_DIR, file + '.pred') + data_test.to_csv(save_path, index=False, sep=' ') \ No newline at end of file