Skip to content

Instantly share code, notes, and snippets.

@eyaltrabelsi

eyaltrabelsi/test_integration_ml_flow.py

Created June 21, 2024 08:16
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save eyaltrabelsi/b4a84ccb3b859ca69090c2ddfbd3529d to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save eyaltrabelsi/b4a84ccb3b859ca69090c2ddfbd3529d to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. eyaltrabelsi created this gist Jun 21, 2024.
    10 changes: 10 additions & 0 deletions test_integration_ml_flow.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,10 @@
    def test_integration_ml_flow():
    input_df = get_production_sample(pct=0.1)
    features_df = feature_engineering(input_df)
    train_df, test_df = split_train_test(features_df)

    model = train_model(train_df)
    onnx_model = export_model(model)

    predictions = inference(model, inference_df)
    assert valid_prediction_histogram(predictions)