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