[ML-35] Restrict printNumericTable to first 10 eigenvalues with first…

… 20 dimensions (#36) * restrict PCA printNumericTable to first 10 eigenvalues with first 20 dimensions * fix ALS printNumericTable
oap-project · Mar 19, 2021 · b4708a4 · b4708a4
1 parent c07d70c
commit b4708a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mllib-dal/src/main/native/ALSDALImpl.cpp b/mllib-dal/src/main/native/ALSDALImpl.cpp
@@ -613,8 +613,8 @@ JNIEXPORT jlong JNICALL Java_org_apache_spark_ml_recommendation_ALSDALImpl_cDALI
 
   std::cout << "\n=== Results for Rank " << rankId << "===\n" << std::endl;
   // std::cout << "Partition ID: " << partitionId << std::endl;
-  printNumericTable(pUser, "User Factors (first 10 rows):", 10);
-  printNumericTable(pItem, "Item Factors (first 10 rows):", 10);
+  printNumericTable(pUser, "User Factors (first 10 rows x 20 columns):", 10, 20);
+  printNumericTable(pItem, "Item Factors (first 10 rows x 20 columns):", 10, 20);
   std::cout << "User Offset: " << getOffsetFromOffsetTable(userOffset) << std::endl;
   std::cout << "Item Offset: " << getOffsetFromOffsetTable(itemOffset) << std::endl;
   std::cout << std::endl;

diff --git a/mllib-dal/src/main/native/PCADALImpl.cpp b/mllib-dal/src/main/native/PCADALImpl.cpp
@@ -125,8 +125,8 @@ JNIEXPORT jlong JNICALL Java_org_apache_spark_ml_feature_PCADALImpl_cPCATrainDAL
     std::cout << "PCA (native): master step took " << duration << " secs" << std::endl;
 
     /* Print the results */
-    printNumericTable(result->get(pca::eigenvalues), "First 10 Eigenvalues:", 10);
-    printNumericTable(result->get(pca::eigenvectors), "First 10 Eigenvectors:", 10);
+    printNumericTable(result->get(pca::eigenvalues), "First 10 eigenvalues with first 20 dimensions:", 10, 20);
+    printNumericTable(result->get(pca::eigenvectors), "First 10 eigenvectors with first 20 dimensions:", 10, 20);
 
     // Return all eigenvalues & eigenvectors