Skip to content

Add deep learning wrappers#36

Merged
fuwiak merged 1 commit intomasterfrom
bober/integrate-keras,-pytorch-with-mcp
May 28, 2025
Merged

Add deep learning wrappers#36
fuwiak merged 1 commit intomasterfrom
bober/integrate-keras,-pytorch-with-mcp

Conversation

@fuwiak
Copy link
Copy Markdown
Owner

@fuwiak fuwiak commented May 28, 2025

Summary

  • add KerasModel, PyTorchModel and PyTorchLightningModel wrappers
  • export wrappers in ML package
  • document Keras and PyTorch support in README
  • allow forwarding metrics via MCP to AI agents
  • add tests for new wrappers

Testing

  • pytest -q

@fuwiak fuwiak merged commit bcfbb19 into master May 28, 2025
3 of 9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant