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

Add post-training verification #16

Open
daanelson opened this issue Jan 27, 2025 · 1 comment
Open

Add post-training verification #16

daanelson opened this issue Jan 27, 2025 · 1 comment

Comments

@daanelson
Copy link
Contributor

daanelson commented Jan 27, 2025

I'd like to add the ability to run a prediction on a trained model and validate that the model has actually learned what we want it to. Good to verify that a trainer not only runs but also trains effectively.

I think the validation would have to live in the context of an individual training. Config design would probably be something like :

train:
  destination: <generated prediction model, e.g. andreasjansson/test-predict. leave
    blank to append '-dest' to the test model>
  destination_hardware: <hardware for the created prediction model, e.g. cpu>
  test_cases:
  - exact_string: <exact string match>
     inputs:
       <input1>: <value1>

    # shiny and new
    post-training inputs:
      <input1>:<value1>
    post-training match: 
      <output1>: <condition1>

unsure if we'd want to move training matching underneath inputs - that'd break compatibility but also be a more reasonable partition.

@andreasjansson
Copy link
Member

Good idea @daanelson!

I think it'd make sense to put trained_model_inputs as the same level as inputs under train, so that each trained test case gets exercised.

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

No branches or pull requests

2 participants