GreNait / technical-thesis

This thesis contains all the work related to it. Python code, models, documentation etc.
0 stars 0 forks source link

model comparison of better model evaluation #29

Open GreNait opened 4 years ago

GreNait commented 4 years ago

Generate different modeld to evaluate their performence with different parameters:

GreNait commented 4 years ago

I generated different models (6 up to now) which are all similar but different in neurons and convolutional layers. The idea is to get a better overview of the capabilities of a model.

GreNait commented 4 years ago

image

GreNait commented 4 years ago

It appears, that something is completly off. I get from the beginning "nan" in the loss function.image

GreNait commented 4 years ago

I made a heavy mistake, with the categorie in the default parameters. I have 3 figures and only provided 2 labels. Hterefore, the nan appeared. I changed that, and the model now starts training with a loss.image

GreNait commented 4 years ago

image

GreNait commented 4 years ago

All three models (excluding non cnns) are able to detect somehow the figures. But they are not perfect. Antman is discovered as batman or vice versa. My guess is, that they all tend to overfitting therefore have problems with the recognition. I will try it again, this time with a dropout rate.

Normally, the first 4 should be a 0 (antman) -> cnn_small image

GreNait commented 4 years ago

Treined the big model with the Dropout, and the detection workded:

image

GreNait commented 4 years ago

I found finally a way to get the steps per epoch right. The imageDtaGenerator has a way to provide that. Just use: "imageDataGenerator2.samples. Thats it. TOTAL_TRAIN can than easily calculated by = self.train_data_generator.samples/self.train_data_generator.batch_size