search

dhruvrajan / tensorflow-keras-java

MIT License
54 stars 9 forks source link

Work on bringing up to date with tensorflow-java #16

Open Sciss opened 3 years ago

Sciss commented 3 years ago

Hi there,

I'm having a look at your project, and attempted to bring it up to date to current API of tensorflow-java. Have a look. Happy to merge back (I can undo using sbt instead of maven). There is one test that currently fails, probably due to wrong extraction of flat 1D array of two-dimensional tensor.

Sciss commented 3 years ago

I'm working now on adding more layers, but switched to Scala because it so much easier to translate the Python code. Would you be interested in checking out my new layer code, and give some feedback or join efforts?

dhruvrajan commented 2 years ago

Hey @Sciss; so sorry I hadn't seen your PR until now! After cleaning up my git notifications I just found it.

If you still happen to be working on this I'm happy to merge this PR; though I imagine using the high-level api's of tensorflow-java these days is the way to go.