Update Rosen_PFN4BO.py

Rosen_PFN4BO.py

@@ -4,7 +4,7 @@ import scipy
 import math
 from sklearn.preprocessing import power_transform, PowerTransformer, StandardScaler
 
-from torchvision.transforms.functional import to_tensor
+# from torchvision.transforms.functional import to_tensor
 from pfns4bo import transformer
 from pfns4bo import bar_distribution
 
@@ -54,7 +54,7 @@ def Rosen_PFN(model_name,
         pt = StandardScaler()
         pt.fit(trained_Y)
         PT_trained_Y = pt.transform(trained_Y)
-        trained_Y = to_tensor(PT_trained_Y).to(torch.float32).reshape(trained_Y.shape)
+        # trained_Y = to_tensor(PT_trained_Y).to(torch.float32).reshape(trained_Y.shape)
     elif trasform_type== 'power':
         pt = PowerTransformer(method="yeo-johnson")
         pt.fit(trained_Y.detach().numpy())