SGML: A Python library for solution-guided machine learning
Researchers have long been concerned with the extrapolation capabilities of machine learning (ML) models, particularly when dealing with insufficient training data. The recently proposed solution-guided machine learning (SGML) method addresses this issue by integrating existing solutions as additional features to supplement limited training data. We have applied this method to solve the strong nonlinearity in nanoindentation and present an approximate solution to the tangential entropic force in an asymmetrical two dimensional bilayer. To make this method more accessible, we developed a user-friendly Python library called SGML, available on GitHub and PyPI. This paper introduces the architecture and functionality of the library, provides a usage example, and discusses its potential impact and applications. • A simple, versatile and user-friendly Python library for implementing the SGML method. • Integrates multiple machine learning models for enhanced functionality. • Supports secondary development for extended customization.
Authors
- Tienchong Chang (ORCID: https://orcid.org/0000-0001-5811-1182)
- Chunchun Dai
- Yang Deng (ORCID: https://orcid.org/0000-0002-8122-5943)
- Ruijin Wang (ORCID: https://orcid.org/0000-0002-6114-3326)
- Yuchen Du (ORCID: https://orcid.org/0000-0002-4723-6884)
- Jiantao Leng (ORCID: https://orcid.org/0000-0002-1371-9351)
Institutions
- Shanghai University (CN)
- Tongji University (CN)
- Zhejiang Lab (CN)
Publication Details
- Journal
- Software Impacts
- Published
- 2024-12-27
- DOI
- https://doi.org/10.1016/j.simpa.2024.100739
- Primary Topic
- Model Reduction and Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China