From 2a3f3aa273abcf747a3e6636427343984ae045d3 Mon Sep 17 00:00:00 2001 From: PawelDopierala Date: Wed, 12 Jun 2024 13:48:32 +0200 Subject: [PATCH] fix sacred --- sacred/helper.py | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 sacred/helper.py diff --git a/sacred/helper.py b/sacred/helper.py new file mode 100644 index 0000000..ffe9422 --- /dev/null +++ b/sacred/helper.py @@ -0,0 +1,9 @@ +import tensorflow as tf + + +def prepare_tensors(df): + Y = df["Price"] + X = df.drop("Price", axis=1) + X_tensor = tf.convert_to_tensor(X, dtype=tf.float32) + Y_tensor = tf.convert_to_tensor(Y, dtype=tf.float32) + return X_tensor, Y_tensor \ No newline at end of file