DIA
/
enose_2025
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

s

main
ifiguero 2025-03-16 05:05:13 -03:00
parent 4788cad0fe
commit 7695340e91
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TrainerClass.py
View File

@ -154,7 +154,7 @@ class eNoseTrainer:
return mse, mae, rmse, optimized_model, model_params return mse, mae, rmse, optimized_model, model_params
def train_and_score_conv1D_v1(self, X_train, X_test, y_train, y_test, epochs=30, num_samples=25): def train_and_score_conv1D_v1(self, X_train, X_test, Y_train, Y_test, epochs=30, num_samples=25):
ray.init(ignore_reinit_error=True) ray.init(ignore_reinit_error=True)
X_train_ref = ray.put(X_train) X_train_ref = ray.put(X_train)
Y_train_ref = ray.put(Y_train) Y_train_ref = ray.put(Y_train)
@ -218,12 +218,12 @@ class eNoseTrainer:
Y_test_pred = best_model.predict(X_test) Y_test_pred = best_model.predict(X_test)
mse_train = mean_squared_error(Y_train, Y_train_pred) mse_train = mean_squared_error(Y_train, Y_train_pred)
mae_test = mean_absolute_error(y_test, Y_test_pred) mae_test = mean_absolute_error(Y_test, Y_test_pred)
mse_test = mean_squared_error(Y_test, Y_test_pred) mse_test = mean_squared_error(Y_test, Y_test_pred)
rmse_test = np.sqrt(mse_test) rmse_test = np.sqrt(mse_test)
# # Calculate evaluation metrics # # Calculate evaluation metrics
mse = mean_squared_error(y_test, y_pred) mse = mean_squared_error(Y_test, y_pred)
rmse = np.sqrt(mse) rmse = np.sqrt(mse)
return mse_train, mae_test, mse_test, rmse_test, best_model, best_config return mse_train, mae_test, mse_test, rmse_test, best_model, best_config