search

nidhaloff / igel

a delightful machine learning tool that allows you to train, test, and use models without writing code
https://igel.readthedocs.io/en/latest/
MIT License
3.09k stars 172 forks source link

Custom CNN Class added #86

Closed Prasanna28Devadiga closed 3 years ago

Prasanna28Devadiga commented 3 years ago

Have created a customizable CNN class as was requested. It supports all the layers that are currently supported by the keras . The image dataset is loaded from a csv file containing the paths and labels in batches .The code has been tested on the MNIST dataset.

nidhaloff commented 3 years ago

@Prasanna28Devadiga you build something very specific to a use case. How is your code dynamic? You are not reading a yaml file, you are not using any parameters that a user could have configured... How is this helpful?

Prasanna28Devadiga commented 3 years ago

I was under the impression that the yaml file is yet to be configured, and we could then just change those two lines that are right now demonstrating a specific use case. So for the yaml file are we going with the syntax you mentioned earlier? If so i'll add it and create a new pr