{"id":17267151,"url":"https://github.com/learnables/torchml","last_synced_at":"2025-04-14T07:53:40.142Z","repository":{"id":169992510,"uuid":"482146078","full_name":"learnables/torchml","owner":"learnables","description":"Scikit-learn implemented with PyTorch","archived":false,"fork":false,"pushed_at":"2023-05-30T20:04:20.000Z","size":24310,"stargazers_count":8,"open_issues_count":16,"forks_count":1,"subscribers_count":4,"default_branch":"master","last_synced_at":"2025-03-27T21:39:02.943Z","etag":null,"topics":["machine-learning","pytorch","scikit-learn"],"latest_commit_sha":null,"homepage":"http://learnables.net/torchml/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/learnables.png","metadata":{"files":{"readme":"README.md","changelog":"CHANGELOG.md","contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null}},"created_at":"2022-04-16T03:35:16.000Z","updated_at":"2025-02-04T10:05:51.000Z","dependencies_parsed_at":"2023-12-31T08:31:42.985Z","dependency_job_id":"9060e443-b800-4e41-ae1b-cd2b3730ab09","html_url":"https://github.com/learnables/torchml","commit_stats":{"total_commits":57,"total_committers":6,"mean_commits":9.5,"dds":0.543859649122807,"last_synced_commit":"77682bd9eda5d2becc47f4d5bad030d427a344ed"},"previous_names":["learnables/torchml"],"tags_count":2,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/learnables%2Ftorchml","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/learnables%2Ftorchml/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/learnables%2Ftorchml/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/learnables%2Ftorchml/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/learnables","download_url":"https://codeload.github.com/learnables/torchml/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":248142903,"owners_count":21054671,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["machine-learning","pytorch","scikit-learn"],"created_at":"2024-10-15T08:09:29.098Z","updated_at":"2025-04-14T07:53:40.106Z","avatar_url":"https://github.com/learnables.png","language":"Python","readme":"\u003cp align=\"center\"\u003e\u003cimg src=\"https://learnables.net/torchml/assets/images/torchml-logo.png\" height=\"120px\" /\u003e\u003c/p\u003e\n\n--------------------------------------------------------------------------------\n\n![Test Status](https://github.com/learnables/torchml/workflows/Testing/badge.svg?branch=master)\n\n`torchml` implements the scikit-learn API on top of PyTorch.\nThis means we automatically get GPU support for scikit-learn and, when possible, differentiability.\n\n## Resources\n\n- GitHub: [github.com/learnables/torchml](http://github.com/learnables/torchml)\n- Documentation: [learnables.net/torchml](http://learnables.net/torchml/)\n- Tutorials: [learnables.net/torchml/tutorials](http://learnables.net/torchml/tutorials/linear_model/)\n- Examples: [learnables.net/torchml/examples](https://github.com/learnables/torchml/tree/master/examples)\n\n## Getting Started\n\n`pip install torchml`\n\n### Minimal Linear Regression Example\n\n~~~python\nimport torchml as ml\n\n(X_train, y_train), (X_test, y_test) = generate_data()\n\n# API closely follows scikit-learn\nlinreg = ml.linear_model.LinearRegression()\nlinreg.fit(X_train, y_train)\nlinreg.predict(X_test)\n~~~\n\n## Changelog\n\nA human-readable changelog is available in the [CHANGELOG.md](./CHANGELOG.md) file.\n\n## Citing\n\nTo cite `torchml` repository in your academic publications, please use the following reference.\n\n\u003e  Sébastien M. R. Arnold, Lucy Xiaoyang Shi, Xinran Gao, Zhiheng Zhang, and Bairen Chen. 2023. \"torchml: a scikit-learn implementation on top of PyTorch\".\n\nYou can also use the following Bibtex entry:\n\n~~~bib\n@misc{torchml,\n  author={Arnold, S{\\'e}bastien M R and Shi, Lucy Xiaoyang and Gao, Xinran and Zhang, Zhiheng and Chen, Bairen},\n  title={torchml: A scikit-learn implementation on top of PyTorch},\n  year={2023},\n  url={https://github.com/learnables/torchml},\n}\n~~~\n","funding_links":[],"categories":[],"sub_categories":[],"project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Flearnables%2Ftorchml","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Flearnables%2Ftorchml","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Flearnables%2Ftorchml/lists"}