You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@texera.apache.org by li...@apache.org on 2026/02/25 00:08:01 UTC
(texera) branch xinyuan-fix-sklearn-testing created (now 08d35242c6)
This is an automated email from the ASF dual-hosted git repository.
linxinyuan pushed a change to branch xinyuan-fix-sklearn-testing
in repository https://gitbox.apache.org/repos/asf/texera.git
at 08d35242c6 init
This branch includes the following new commits:
new 08d35242c6 init
The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
(texera) 01/01: init
Posted by li...@apache.org.
This is an automated email from the ASF dual-hosted git repository.
linxinyuan pushed a commit to branch xinyuan-fix-sklearn-testing
in repository https://gitbox.apache.org/repos/asf/texera.git
commit 08d35242c654a29be36b4c89108248374d75b601
Author: Xinyuan Lin <xi...@uci.edu>
AuthorDate: Tue Feb 24 16:07:53 2026 -0800
init
---
.../texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala
index 4c7af2db98..df7d933665 100644
--- a/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala
+++ b/common/workflow-operator/src/main/scala/org/apache/texera/amber/operator/sklearn/testing/SklearnTestingOpDesc.scala
@@ -69,7 +69,7 @@ class SklearnTestingOpDesc extends PythonOperatorDescriptor {
| table = Table(self.data)
| Y = table[$target]
| X = table.drop($target, axis=1)
- | predictions = model.predict(X)
+ | predictions = model.predict(X.squeeze())
| if $isRegressionStr:
| tuple_["R2"] = r2_score(Y, predictions)
| tuple_["RMSE"] = root_mean_squared_error(Y, predictions)