-
Notifications
You must be signed in to change notification settings - Fork 111
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
Test SelectKBest + StandardScaler pipeline #1156
Conversation
[("input", FloatTensorType([None, X.shape[1]]))], | ||
target_opset=TARGET_OPSET, | ||
) | ||
self.assertTrue(model_onnx is not None) |
Check notice
Code scanning / CodeQL
Imprecise assert Note test
38d4e19
to
6bff77d
Compare
@xadupre I fixed the issue (just for All selectors such as What bothers me is that in the end the So my two questions are:
|
You should remove the register and run |
No |
X = np.array( | ||
[[1, 2, 3, 1], [0, 3, 1, 4], [3, 5, 6, 1], [1, 2, 1, 5]], dtype=np.int64 | ||
) | ||
y = np.array([0, 1, 0, 1]) | ||
model.fit(X, y) | ||
|
||
model_onnx = convert_sklearn( | ||
model, |
Check notice
Code scanning / CodeQL
Imprecise assert Note test
Signed-off-by: Pierre Bartet <[email protected]>
Signed-off-by: Pierre Bartet <[email protected]>
Signed-off-by: Pierre Bartet <[email protected]>
Signed-off-by: Pierre Bartet <[email protected]>
Signed-off-by: Pierre Bartet <[email protected]>
a4af89e
to
1dad533
Compare
@xadupre done |
When$k$ smaller than the number of features, the conversion fails:
SelectKBest
is used before aStandardScaler
inside a pipeline withThis PR fixes this problem by using the
concat
shape calculator for selectors and adding a correct one instead.