Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: SparkML StandardScaler conversion fails when withStd or withMean is set to true #555

Merged
merged 2 commits into from
Jun 5, 2022

Conversation

memoryz
Copy link
Contributor

@memoryz memoryz commented May 28, 2022

Signed-off-by: Jason Wang jasowang@microsoft.com

@memoryz
Copy link
Contributor Author

memoryz commented May 28, 2022

Fixing #540 .

… is set to true

Signed-off-by: Jason Wang <jasowang@microsoft.com>
@memoryz memoryz force-pushed the jasowang/iss540 branch from 4b93644 to 650c4d6 Compare May 31, 2022 17:34
@memoryz
Copy link
Contributor Author

memoryz commented May 31, 2022

@xadupre This is ready for review.

@memoryz
Copy link
Contributor Author

memoryz commented Jun 4, 2022

@xadupre ping.

@@ -75,7 +75,7 @@ def test_standard_scaler(self):
(1, Vectors.dense([2.0, 1.1, 1.0]),),
(2, Vectors.dense([3.0, 10.1, 3.0]),)
], ["id", "features"])
scaler = StandardScaler(inputCol='features', outputCol='scaled_features')
scaler = StandardScaler(inputCol='features', outputCol='scaled_features', withStd=True, withMean=True)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No need to test other values of withStd, withMean?

@xadupre xadupre merged commit 4e1fe87 into onnx:main Jun 5, 2022
@memoryz memoryz deleted the jasowang/iss540 branch June 8, 2022 01:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants