search

miladmozafari / SpykeTorch

High-speed simulator of convolutional spiking neural networks with at most one spike per neuron.
GNU General Public License v3.0
381 stars 100 forks source link

dataset #9

Closed guzh970630 closed 2 years ago

guzh970630 commented 3 years ago

hi~, I want to ask a question. I am very interested in your network, but I want to use my dataset, which is a csv file. However, I don’t know what your MNIST input format looks like, numpy, list, or tensor? If I want to prepare it myself, what kind of format do I need? Because my data set is not a picture data set, but can be directly input into numpy, list or tensor.

miladmozafari commented 2 years ago

Hello, First of all, I apologize for the very late response. For some reason, I missed your comment! Thank you for your interest in SpykeTorch. For the dataset, you need to convert your data into tensors before giving them to the network. Working with SpykeTorch models is pretty the same as PyTorch. For MNIST, I used torchvision library which implements an adapter for it. However, for your CSV file, you may need to implement your own dataset.