how to train custom images dataset

[VYRION-Ai]

i have images dataset of two classes , i need to train Qnn on my dataset ,
how i can do that ?

[GenericP3rson]

Hi! If you check out some the files in examples, we have some examples of a QNN trained over MNIST (examples/mnist) and using a quavolution network also for MNIST (examples/quanvolution).

For your own data, you’ll probably just need to modify the dataloader, but the models can more or less be the same, perhaps save some hyperparameter tuning.

Hopefully this can be a helpful start, and let me know if you run into any issues along the way and would be happy to help debug!

[VYRION-Ai]

@GenericP3rson i tried a lot but i did not success , and also i work with 2 class not 4 , can you please give me little hints

[GenericP3rson]

Interesting, if possible, could you share some of your code / specify what issues you’ve been running into?

I know this sort of thing has been requested before, so I can try to spin up an example when I have the bandwidth. For now though, I can help debug!