refactor(data_loader): cast "Target" column to float32 in get_dataset…

… function

src/dataset_loader.py

@@ -233,13 +233,15 @@ def get_dataset(file_path, batch_size=2, shuffle_size=100, shuffle=False):
             f"Required columns {required_columns} not present in the DataFrame."
         )
 
+    dataframe = dataframe.with_columns(dataframe["Target"].cast(pl.Float32))
     features_df = dataframe["Log"].to_numpy()
     target_df = dataframe["Target"].to_numpy()
 
     dataset = tf.data.Dataset.from_tensor_slices((features_df, target_df))
 
     if shuffle:
         dataset = dataset.shuffle(shuffle_size)
+    # dataset = dataset.map(convert_label_to_float)
     dataset = dataset.batch(batch_size).prefetch(buffer_size=tf.data.AUTOTUNE)
     return dataset