Spaces:

FranciscoLozDataScience
/

uhi_resnet_model

Sleeping

FranciscoLozDataScience commited on Mar 17

Commit

bc58ccd

1 Parent(s): 2225e11

remove a step from predict func

Files changed (1) hide show

model.py CHANGED Viewed

@@ -122,9 +122,6 @@ class UhiModel:
         # Scale the input data
         X_scaled = self.scale(X_processed)
-        # Ensure the scaled data is 2D
-        X_scaled = X_scaled.reshape(1, -1)
         # Make prediction
         y_pred = self.model.predict(X_scaled)

         # Scale the input data
         X_scaled = self.scale(X_processed)
         # Make prediction
         y_pred = self.model.predict(X_scaled)