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

Add code for utilizing GPU #59

Open
wants to merge 4 commits into
base: master
Choose a base branch
from

Conversation

mainguyenanhvu
Copy link

I add one parameter for using TensorFlow Decision Forests (TF-DF). This package can help us to utilize GPU when we install tensorflow-gpu.

I believe that it could enhance the performance of ppscore when we have a large number of features by making use of GPUs.

@FlorianWetschoreck
Copy link
Collaborator

Thank you a lot for your PR! I will have a look at it in the new year.

@mainguyenanhvu
Copy link
Author

mainguyenanhvu commented Dec 31, 2021

I run test_calculation.py for my code. However, it did pass test for random_seed. I do not know why even when I printed the results to check. It illustrated they were the same.

python3 tests/test_calculation.py 
WARNING:root:TF Parameter Server distributed training not available (this is expected for the pre-build release).
{'x': 'x', 'y': 'y', 'ppscore': 0.6554396626978072, 'case': 'regression', 'is_valid_score': True, 'metric': 'mean absolute error', 'baseline_score': 1.0074503071153915, 'model_score': 0.347127417634877, 'model': DecisionTreeRegressor()}
{'x': 'x', 'y': 'y', 'ppscore': 0.6554396626978072, 'case': 'regression', 'is_valid_score': True, 'metric': 'mean absolute error', 'baseline_score': 1.0074503071153915, 'model_score': 0.347127417634877, 'model': DecisionTreeRegressor()}
False
Traceback (most recent call last):
  File "tests/test_calculation.py", line 335, in <module>
    main()
  File "tests/test_calculation.py", line 326, in main
    test_score()
  File "tests/test_calculation.py", line 176, in test_score
    df, "x", "y", random_seed=1
AssertionError

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