kstaats / karoo_gp

A Genetic Programming platform for Python with TensorFlow for wicked-fast CPU and GPU support.
Other
157 stars 61 forks source link

Tf2 update #97

Closed granawkins closed 1 year ago

granawkins commented 1 year ago

This PR implements tensorflow 2 and in-place evaluation of trees, as detailed in #73.

Hiccup: two tests were failing:

However, the results for the other 42 tests are identical, including the multi-generation runs for 'classification', 'regression' and the other 'matching' tasks. This suggests it's something to do with the matching kernel specifically, perhaps related to the build_fittest_dict function. We'd planned on deprecating the matching kernel anyway, so this might not be important.

Anyway, I'll look into it more later (out of time for now) and try to understand the issue fully before committing.