crestonbunch / tbcnn

Efficient tree-based convolutional neural networks in TensorFlow
MIT License
150 stars 44 forks source link

Implement tbcnn on my own dataset #16

Open QiushiSun opened 3 years ago

QiushiSun commented 3 years ago

Hi, I'm trying to Implement tbcnn on my own dataset, but having some trouble about preprocessing my source code to the .pkl file named algorithm.pkl. I would be grateful if you could provide me a method to convert the source code to the .pkl file in the project, thanks.

bdqnghi commented 3 years ago

hi, you can refer to my improved version here: https://github.com/bdqnghi/tbcnn-tensorflow, which includes more details instruction, and reproduce the results of the original paper.