mert-kurttutan
Made the colors of the graph parametrizable
This pull request adds the color scheme as a member variable to ComputationGraph. By this means, the color scheme of the graph can be changed by setting it via a parameter in the function draw_graph.
A test was added to ensure the new feature works.
This PR resolves #111
The latest updates on your projects. Learn more about Vercel for Git ↗︎
| Name | Status | Preview | Comments | Updated (UTC) |
|---|---|---|---|---|
| torchview | ✅ Ready (Inspect) | Visit Preview | 💬 Add feedback | Nov 4, 2024 10:57am |
![vercel[bot] avatar](https://avatars.githubusercontent.com/in/8329?v=4 "Published by a pub.dev verified publisher"){kind=small}
This seems useful and simple enough to implement. Can you also provide a larger (and more practically used example, something like BERT) where we can see the benefit of this (vs previous solution)?.
We should also put some sort of validation for color scheme input, making sure that correct keys are passed and color names. Correct keys is simple. What about color names? Does graphviz provide validation or error when incorrect color names are passed ?