Skip to content

Conversation

@antoniofilipovic
Copy link
Contributor

@antoniofilipovic antoniofilipovic commented Nov 8, 2022

Description

This PR includes three bugfixes:

  • training throws an error after calling the reset method
  • training not working if class labels don't start with 0
  • potential bug for early stopping on the first iteration if val_loss > 100

Pull request type

  • Bugfix

######################################

Reviewer checklist (the reviewer checks this part)

Module/Algorithm

  • Core algorithm/module implementation
  • Query module implementation
  • Unit tests
  • End-to-end tests
  • Code documentation
  • README short description
  • Documentation on memgraph/docs

######################################

@antoniofilipovic antoniofilipovic changed the title [main < T469-FL] Fix reset on node classification [main < T469-FL] Fix reset, class label and early stopping bugs on node classification module Nov 9, 2022
Copy link
Contributor

@antepusic antepusic left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

All looks good so far 🚀

@antoniofilipovic antoniofilipovic added the status: ready PR is ready for review label Nov 10, 2022
@antoniofilipovic antoniofilipovic merged commit 4a589fe into main Nov 14, 2022
@antoniofilipovic antoniofilipovic deleted the T469-FL-fix-reset-node-classification branch November 14, 2022 13:32
@antepusic antepusic added status: ship it PR approved and removed status: ready PR is ready for review labels Nov 14, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

status: ship it PR approved

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

4 participants