search

tensorflow / decision-forests

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.
Apache License 2.0
663 stars 110 forks source link

Training Duration of Penguin Example #31

Closed JulianHBuecher closed 3 years ago

JulianHBuecher commented 3 years ago

Hello all together, I have a short question regarding the training time needed by this specific model.

For digging into the material I used the example from TensorFlow Website with the Penguin Data and started the training on my Linux Laptop with a NVIDIA GeForce GTX 1050 Ti with GPU support enabled.

Now I am wondering why the model takes more than a hour for only the training of 300 rows of data with 5 features or so...

Have anyone a benchmark value? I would really appreciate your help guys. Best regards Julian

JulianHBuecher commented 3 years ago

Additional Information: The Training Run above runs as a Python Script in my console.

For comparison I tried the Jupyter Notebook available from the tutorial site. Inside the notebook the model training works within seconds... Don't understand it :D Why it works inside a notebook and not as a normal script? Here is my code as reference: https://github.com/JulianHBuecher/Bachelor-Thesis/blob/main/ML.Proxy.ModelTrainer.Python/TensorFlow.DF/randomforest_Penguin.py