mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-23 17:48:23 +00:00
*Issue #, if available:* #354 *Description of changes:* This PR adds `Chronos2Pipeline.embed` to enable users to extract embeddings from the last encoder layer in an easy way. The API and behavior is similar to what Chronos and Chronos-Bolt provides. By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. |
||
|---|---|---|
| .. | ||
| dummy-chronos-bolt-model | ||
| dummy-chronos-model | ||
| dummy-chronos2-model | ||
| __init__.py | ||
| test_chronos.py | ||
| test_chronos2.py | ||
| test_chronos_bolt.py | ||
| test_utils.py | ||
| util.py | ||