text: Title: [BUG] xdeepfm error in AzureML test
  Body: ### Description
  <!--- Describe your issue/bug/request in detail -->
  ```
      @pytest.mark.gpu
      @pytest.mark.notebooks
      @pytest.mark.integration
      @pytest.mark.parametrize(
          "syn_epochs, criteo_epochs, expected_values, seed",
          [
              (
                  15,
                  10,
                  ***
                      "res_syn": ***"auc": 0.9716, "logloss": 0.699***,
                      "res_real": ***"auc": 0.749, "logloss": 0.4926***,
                  ***,
                  42,
              )
          ],
      )
      def test_xdeepfm_integration(
          notebooks,
          output_notebook,
          kernel_name,
          syn_epochs,
          criteo_epochs,
          expected_values,
          seed,
      ):
          notebook_path = notebooks["xdeepfm_quickstart"]
          pm.execute_notebook(
              notebook_path,
              output_notebook,
              kernel_name=kernel_name,
              parameters=dict(
                  EPOCHS_FOR_SYNTHETIC_RUN=syn_epochs,
                  EPOCHS_FOR_CRITEO_RUN=criteo_epochs,
                  BATCH_SIZE_SYNTHETIC=1024,
                  BATCH_SIZE_CRITEO=1024,
                  RANDOM_SEED=seed,
              ),
          )
          results = sb.read_notebook(output_notebook).scraps.dataframe.set_index("name")[
              "data"
          ]
      
          for key, value in expected_values.items():
  >           assert results[key]["auc"] == pytest.approx(value["auc"], rel=TOL, abs=ABS_TOL)
  E           assert 0.5131 == 0.9716 ± 9.7e-02
  E             comparison failed
  E             Obtained: 0.5131
  E             Expected: 0.9716 ± 9.7e-02
  ```
  
  ### In which platform does it happen?
  <!--- Describe the platform where the issue is happening (use a list if needed) -->
  <!--- For example: -->
  <!--- * Azure Data Science Virtual Machine. -->
  <!--- * Azure Databricks.  -->
  <!--- * Other platforms.  -->
  
  ### How do we replicate the issue?
  <!--- Please be specific as possible (use a list if needed). -->
  <!--- For example: -->
  <!--- * Create a conda environment for pyspark -->
  <!--- * Run unit test `test_sar_pyspark.py` with `pytest -m 'spark'` -->
  <!--- * ... -->
  See https://github.com/microsoft/recommenders/actions/runs/3459763061/jobs/5775521889
  
  
  ### Expected behavior (i.e. solution)
  <!--- For example:  -->
  <!--- * The tests for SAR PySpark should pass successfully. -->
  
  ### Other Comments
  