fix sacred
This commit is contained in:
parent
1c0126376e
commit
2a3f3aa273
9
sacred/helper.py
Normal file
9
sacred/helper.py
Normal file
@ -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
|
Loading…
Reference in New Issue
Block a user