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

Feature/experiment tracking #98

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

Conversation

adamkells
Copy link
Contributor

@adamkells adamkells commented Nov 8, 2024

Description

The purpose of this PR is to add experiment tracking to Healthchain.

Related Issue

#57

Changes Made

A new class ExperimentTracker is created for usage inside the sandbox decorator. This automatically tracks any metadata about experiments run and saves the data to a local sql database.

A few additional bug fixes were made for LLM integration tests which were failing.

Testing

The summarization pipeline example in the cookbook was modified to use the experiment tracker and run locally to confirm that the expected behaviour occurs

Checklist

  • I have read the contributing guidelines
  • I have performed a self-review of my own code
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes

Additional Notes

@adamkells adamkells changed the base branch from feature/pipeline-implementations to main December 4, 2024 10:07
@adamkells adamkells marked this pull request as ready for review December 4, 2024 10:08
Copy link
Member

@jenniferjiangkells jenniferjiangkells left a comment

Choose a reason for hiding this comment

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

lgtm. Just out of curiosity do you have an example of what the experiment data looks like in a real example? Maybe can put it in docs.

@adamkells
Copy link
Contributor Author

Screenshot from 2024-12-07 22-42-04
Screenshot from 2024-12-07 22-42-18

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.

2 participants