Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement FNN training agent #34

Open
wants to merge 10 commits into
base: main
Choose a base branch
from

Conversation

robilkot
Copy link
Collaborator

@robilkot robilkot commented Nov 7, 2024

Trainer agent is now working.

Capabilities

Specification of FNN structure in kb

image

Specification of preproccessed csv-datasets in kb

image

Specification of training parameters in kb

image

Given above parameters, agent is able to build tensorflow.keras FNN according to provided structure and train it on given dataset. Input and output data shapes are identified automatically. Trained model is then stored in filesystem, as well as linked to FNN structure in database, allowing for future interpretation

Example output during model training:

FNN structure per-layer is shown, as well as training state on each epoch

image

MNIST Fashion dataset is used as example.

Using training agent

  1. Formalize desired FNN structure by providing layers with their sizes and activation functions in ordered way
  2. Provide dataset: specify filepath and labels column
  3. Provide desired training parameters: include learning rate, batch size and number of epochs
  4. Create edge from 'question_initiated' node to action node
  5. Wait for appearence of link containing path to file with trained keras model

Trained models are stored at kb\fnn_processing_module\trained_models

@robilkot robilkot force-pushed the dev_FnnProcessingModule branch 2 times, most recently from 9c44f95 to 63c45f5 Compare November 9, 2024 19:35
@robilkot
Copy link
Collaborator Author

robilkot commented Nov 9, 2024

Moved docker changes to #35

@robilkot robilkot force-pushed the dev_FnnProcessingModule branch from 63c45f5 to d26fb68 Compare November 9, 2024 19:55
@robilkot robilkot force-pushed the dev_FnnProcessingModule branch from d26fb68 to b02f625 Compare November 10, 2024 13:59
@KovalM KovalM self-requested a review December 8, 2024 21:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant