From fc689383d4cbe69cca56c3d8c5a9f07b48742095 Mon Sep 17 00:00:00 2001 From: straile Date: Sat, 9 Nov 2024 16:26:00 +0100 Subject: [PATCH] feat: remove sklearn k means --- .../AI_Python_functions/keras_AI_surrogate.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/src/Chemistry/SurrogateModels/AI_Python_functions/keras_AI_surrogate.py b/src/Chemistry/SurrogateModels/AI_Python_functions/keras_AI_surrogate.py index b37bcd708..5f857338a 100644 --- a/src/Chemistry/SurrogateModels/AI_Python_functions/keras_AI_surrogate.py +++ b/src/Chemistry/SurrogateModels/AI_Python_functions/keras_AI_surrogate.py @@ -1,16 +1,10 @@ import tensorflow as tf import numpy as np -from sklearn.cluster import KMeans import os os.environ["TF_XLA_FLAGS"] = "--tf_xla_cpu_global_jit" os.environ["XLA_FLAGS"] = "--xla_gpu_cuda_data_dir=" + cuda_dir -def k_means(data, k=2, tol=1e-6): - kmeans = KMeans(n_clusters=k, tol=tol) - labels = kmeans.fit_predict(data) - return labels - def initiate_model(model_file_path): print("AI: Model loaded from: " + model_file_path, flush=True) model = tf.keras.models.load_model(model_file_path)