Viewing a single comment thread. View all comments

manojs t1_j5wbyum wrote

With such a small dataset, you should use pre-existing classification models most similar to your data (search huggingface), and then re-train just the last layer or last couple of layers ("freeze" all the prior layers). And yes you can use the data augmentation suggestion but if you build the entire network from scratch it will be challenging to get good results. AKA "transfer learning".

1