-
Notifications
You must be signed in to change notification settings - Fork 30
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
TypeError: __init__() got an unexpected keyword argument 'maxIter' #365
Comments
Thank you Johnny for reporting this issue! Logistic Regression requires cuml 23.08 or above. So you can try upgrading cuml to 23.08. Note only fit is supported currently. transform is under review: #363 The ci has been upgraded to use cuml 23.08, but other parts (e.g. installation instructions) of spark rapids ml still use 23.06. The whole spark rapids ml should be upgraded to cuml 23.08 after the cuml 23.08 is officially released. |
The cuml of my test enviroment is 23.08 already. please get the detail in following. (base) root@93b04b04d66d:/usr/workspace/spark-rapids-ml/python# pip list|grep cuda |
Get it. Spark rapids ml does not support init arguments (e.g. maxIter) yet. Needs this PR #363 merged. |
environment:
python3.9 , linux, ubuntu22.04, spark340, spark-rapids-ml branch23.08
issue:
Hit following error when run statement "lr = LogisticRegression(num_workers=gpu_number, maxIter=10, regParam=0.01)" when use spark_rapids_ml.classification.LogisticRegression
in contrast, run statement "lr = LogisticRegression(maxIter=10, regParam=0.01)" successful when use pyspark.ml.classification.LogisticRegression.
seems like this api is not compatiblity , so file this issue to track.
error:
def test_e2e_simple(gpu_number: int) -> None:
"""e2e general training and transforming test"""
gpu_number = min(gpu_number, 2)
E TypeError: init() got an unexpected keyword argument 'maxIter'
testsuites/test_logistic_regression.py:73: TypeError
======================================== short test summary info =========================================
FAILED testsuites/test_logistic_regression.py::test_e2e_simple - TypeError: init() got an unexpected keyword argument 'maxIter'
====================================== 1 failed, 8 passed in 4.67s =======================================
The text was updated successfully, but these errors were encountered: