search

parthsarthi03 / raptor

The official implementation of RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrieval
https://arxiv.org/abs/2401.18059
MIT License
878 stars 126 forks source link

Support for seeing Created Tree ? #36

Open jalotra opened 4 months ago

jalotra commented 4 months ago

hey fellas, was doing a POC run on top of this, it works great with default configs ! so next step after POC is explaining the flow to leads, so wrote a simple script to display the generated tree using code, have written a simple script that makes it much more meaningful to look at.

this is a sample gist to do this : https://gist.github.com/jalotra/1b1d640ec5233657cfd9121a8ac419b4

example image :

Screenshot 2024-05-10 at 1 00 13 PM

lmk if people want something like this, would raise a PR as well

Giustino98 commented 4 months ago

Hi @jalotra , very interesting stuff! I have tried to run your script, but it seems there are some imports that I can't find. Could you please provide the import scripts to run the algorithm? Thank you so much :)

parthsarthi03 commented 4 months ago

This would be great to have in the repo! Please feel free to file a PR for this and assign me to review it!

jalotra commented 4 months ago

ack @parthsarthi03 will raise a PR for this

jalotra commented 4 months ago

hey @Giustino98 you can use this #40 to fiddle with this, have raised a PR.