Skip to content

GitLab

Merged
Created by Oliver Kirsebom@kirsebomOwner2 of 2 tasks completed2/2 tasks

Refactor neural network

This merge request provides the following improvements/enhancements:

  1. The class CNNWhale is split into a parent class called DataHandler, and a child class called CNNWhale, where DataHandler provides generic implementations of methods related to data handling (set, add, get training/validation/test data).

  2. Some method names have been simplified (e.g. create_net_structure -> create, save_model -> save, etc.)

  3. A new neural network class called 'EDTCN' (Encoder-Decoder Temporal Network) has been implemeted using keras methods from the Tensorflow library.

  4. A new class called TrainingDataProvider has also been implemented. This class facilitates an active learning process, in which extra emphasis is placed on the training examples that yield low confidence predictions or wrong predictions.

Things still to be done:

  1. Implement more unit tests of TrainingDataProvider
  2. Finish implementation of EDTCN (including unit tests)
Edited by Oliver Kirsebom