omnigenome 0.3.1a0__py3-none-any.whl → 0.4.0a0__py3-none-any.whl

omnigenome-0.3.1a0.dist-info/METADATA

@@ -1,224 +0,0 @@
-Metadata-Version: 2.4
-Name: omnigenome
-Version: 0.3.1a0
-Summary: OmniGenome: A comprehensive toolkit for genome analysis.
-Home-page: https://github.com/yangheng95/OmniGenBench
-Author: Yang, Heng
-Author-email: hy345@exeter.ac.uk
-License: Apache-2.0
-Platform: Windows
-Platform: Linux
-Platform: Mac OS-X
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Science/Research
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Operating System :: OS Independent
-Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: findfile>=2.0.0
-Requires-Dist: autocuda>=0.16
-Requires-Dist: metric-visualizer>=0.9.6
-Requires-Dist: termcolor
-Requires-Dist: gitpython
-Requires-Dist: torch>=2.6.0
-Requires-Dist: pandas
-Requires-Dist: viennarna
-Requires-Dist: scikit-learn
-Requires-Dist: accelerate
-Requires-Dist: transformers>=4.46.0
-Requires-Dist: packaging
-Requires-Dist: peft
-Requires-Dist: dill
-Requires-Dist: accelerate
-Provides-Extra: dev
-Requires-Dist: dill; extra == "dev"
-Requires-Dist: pytest; extra == "dev"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license
-Dynamic: license-file
-Dynamic: platform
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-![favicon.png](asset/favicon.png)
-
-<h3 align="center">OmniGenBench is a modular benchmarking platform for evaluating genomic foundation models (GFMs) across diverse tasks like RNA structure prediction, gene function classification, and multi-species generalization.</h3>
-
-<div align="center">
-
-  <a href="https://omnigenbenchdoc.readthedocs.io/en/latest/">
-    <img src="https://img.shields.io/readthedocs/omnigenbench?logo=readthedocs&logoColor=white" alt="Documentation Status" />
-  </a>
-
-  <a href="https://pypi.org/project/omnigenome/">
-    <img src="https://img.shields.io/pypi/v/omnigenome?color=blue&label=PyPI" alt="PyPI" />
-  </a>
-
-  <a href="https://pepy.tech/project/omnigenome">
-    <img src="https://static.pepy.tech/badge/omnigenome" alt="PyPI Downloads" />
-  </a>
-
-  <a href="https://pypi.org/project/omnigenbench/">
-    <img src="https://img.shields.io/pypi/pyversions/omnigenome" alt="Python Version" />
-  </a>
-
-  <a href="https://github.com/yangheng95/omnigenome/blob/main/LICENSE">
-    <img src="https://img.shields.io/github/license/yangheng95/omnigenome" alt="License" />
-  </a>
-
-</div>
-<h3 align="center">
-  <a href="#installation">📦 Installation</a>
-  <span> · </span>
-  <a href="#quick-start">🚀 Getting Started</a>
-  <span> · </span>
-  <a href="#supported-models">🧬 Model Support</a>
-  <span> · </span>
-  <a href="#benchmarks">📊 Benchmarks </a>
-  <span> · </span>
-  <a href="#tutorials">🧪 Application Tutorials</a>
-  <span> · </span>
-  <a href="https://arxiv.org/pdf/2505.14402">📚 Paper</a>
-</h3>
-
-
-## 🔍 What You Can Do with OmniGenBench?
-
-- 🧬 **Benchmark effortlessly** — Run automated and reproducible evaluations for genomic foundation models  
-- 🧠 **Understand your models** — Explore interpretability across diverse tasks and species  
-- ⚙️ **Run tutorials instantly** — Use click-to-run guides for genomic sequence modeling  
-- 🚀 **Train and infer efficiently** — Accelerated workflows for training and inference on GFMs  
-
-## Installation
-
-### Requirements
-Before installing OmniGenome, you need to install the following dependencies:
-- Python 3.10+
-- PyTorch 2.5+
-- Transformers 4.46.0+
-
-### PyPI Installation
-To install OmniGenome, you can use pip:
-```bash
-pip install omnigenome -U
-```
-
-### Source Installation
-Or you can clone the repository and install it from source:
-```bash
-git clone https://github.com/yangheng95/OmniGenBench.git
-cd OmniGenBench
-pip install -e .
-```
-
-## Quick Start
-`OmniGenBench is available for diverse models and benchmark suites, please refer to the  following sections for more details.`
-### Auto-benchmark via CLI
-The following command will download the model from the Hugging Face model hub and run the benchmark on the RGB benchmark:
-```bash
-autobench --model_name_or_path "yangheng/OmniGenome-186M" --benchmark "RGB" --trainer accelerate
-```
-You can find a visualization of AutoBench [here](asset/AutoBench.gif).
-
-
-### Auto-benchmark via Python API
-Or you can use the following python code to run the auto-benchmark:
-```python
-from omnigenome import AutoBench
-gfm = 'LongSafari/hyenadna-medium-160k-seqlen-hf'
-# benchmark could be "RGB", "GB", "PGB", "GUE", which will be downloaded from the Hugging Face model hub
-benchmark = "RGB"
-bench_size = 8
-seeds = [0, 1, 2, 3, 4]
-bench = AutoBench(benchmark=benchmark, model_name_or_path=gfm, overwrite=False)
-bench.run(autocast=False, batch_size=bench_size, seeds=seeds)
-```
-You can find an example of AutoBench via Python API [here](examples/autobench/AutoBench_Tutorial.ipynb).
-
-## Supported Models
-
-
-OmniGenBench provides plug-and-play evaluation for over **30 genomic foundation models**, covering both **RNA** and **DNA** modalities. The following are highlights:
-
-| Model          | Params | Pre-training Corpus                        | Highlights                                          |
-|----------------|--------|--------------------------------------------|-----------------------------------------------------|
-| **OmniGenome** | 186M   | 54B plant RNA+DNA tokens                   | Multi-modal, structure-aware encoder                |
-| **Agro-NT-1B** | 985M   | 48 edible-plant genomes                    | Billion-scale DNA LM w/ NT-V2 k-mer vocab           |
-| **RiNALMo**    | 651M   | 36M ncRNA sequences                        | Largest public RNA LM; FlashAttention-2             |
-| **DNABERT-2**  | 117M   | 32B DNA tokens, 136 species (BPE)          | Byte-pair encoding; 2nd-gen DNA BERT                |
-| **RNA-FM**     | 96M    | 23M ncRNA sequences                        | High performance on RNA structure tasks             |
-| **RNA-MSM**    | 96M    | Multi-sequence alignments                  | MSA-based evolutionary RNA LM                       |
-| **NT-V2**      | 96M    | 300B DNA tokens (850 species)              | Hybrid k-mer vocabulary                             |
-| **HyenaDNA**   | 47M    | Human chromosomes                          | Long-context autoregressive model (1Mb)             |
-| **SpliceBERT** | 19M    | 2M pre-mRNA sequences                      | Fine-grained splice-site recognition                |
-| **Caduceus**   | 1.9M   | Human chromosomes                          | Ultra-compact DNA LM (RC-equivariant)               |
-| **RNA-BERT**   | 0.5M   | 4,000+ ncRNA families                      | Small BERT with nucleotide masking                  |
-| *...and more*  | —      | See Appendix E of the paper                | Includes PlantRNA-FM, UTR-LM, MP-RNA, CALM, etc.    |
-
-## Benchmarks
-
-OmniGenBench supports five curated benchmark suites covering both **sequence-level** and **structure-level** genomics tasks across species.
-
-| Suite        | Focus                       | #Tasks / Datasets        | Sample Tasks                                         |
-|--------------|-----------------------------|--------------------------|------------------------------------------------------|
-| **RGB**      | RNA structure + function    | 12 tasks (SN-level)      | RNA secondary structure, SNMR, degradation prediction |
-| **BEACON**   | RNA (multi-domain)          | 13 tasks                 | Base pairing, mRNA design, RNA contact maps         |
-| **PGB**      | Plant long-range DNA        | 7 categories             | PolyA, enhancer, chromatin access, splice site      |
-| **GUE**      | DNA general tasks           | 36 datasets (9 tasks)    | TF binding, core promoter, enhancer detection       |
-| **GB**       | Classic DNA classification  | 9 datasets               | Human/mouse enhancer, promoter variant classification|
-
-
-## Tutorials
-
-### RNA Design
-RNA design is a fundamental problem in synthetic biology,
-where the goal is to design RNA sequences that fold into a target structure.
-In this demo, we show how to use OmniGenome to design RNA sequences 
-that fold into a target structure using a pre-trained model.
-The tutorials of RNA Design Demo can be found in [RNA_Design_Tutorial.ipynb](examples/rna_design/RNA_Design_Tutorial.ipynb).
-
-You can find a visual example of RNA Design [here](asset/RNA_Design.gif).
-
-### RNA Secondary Structure Prediction
-
-RNA secondary structure prediction is a fundamental problem in computational biology,
-where the goal is to predict the secondary structure of an RNA sequence.
-In this demo, we show how to use OmniGenome to predict the secondary structure of RNA sequences using a pre-trained model.
-The tutorials of RNA Secondary Structure Prediction can be found in
-[Secondary_Structure_Prediction_Tutorial.ipynb](examples/rna_secondary_structure_prediction/Secondary_Structure_Prediction_Tutorial.ipynb).
-
-You can find a visual example of RNA Secondary Structure Prediction [here](asset/RNA_Structure_Prediction.gif).
-
-### More Tutorials
-Please find more usage tutorials in [examples/tutorials](examples/tutorials).
-
-## Citation
-```bibtex
-@article{yang2024omnigenbench,
-      title={OmniGenBench: A Modular Platform for Reproducible Genomic Foundation Models Benchmarking}, 
-      author={Heng Yang and Jack Cole, Yuan Li, Renzhi Chen, Geyong Min and Ke Li},
-      year={2024},
-      eprint={https://arxiv.org/abs/2505.14402},
-      archivePrefix={arXiv},
-      primaryClass={q-bio.GN},
-      url={https://arxiv.org/abs/2505.14402}, 
-}
-```
-## License
-OmniGenBench is licensed under the Apache License 2.0. See the LICENSE file for more information.
-
-
-## Contribution
-We welcome contributions to OmniGenBench! If you have any ideas, suggestions, or bug reports, please open an issue or submit a pull request on GitHub.

omnigenome-0.3.1a0.dist-info/RECORD

@@ -1,78 +0,0 @@
-omnigenome/__init__.py,sha256=2tXjLkxTWA1Akx9iks04X-JZckhVmkIEPDO5wKTucF8,9897
-omnigenome/auto/__init__.py,sha256=UhcuYy43WsR7IowjajlcGwNVFFFDaufl8KqtNDmVqz0,97
-omnigenome/auto/auto_bench/__init__.py,sha256=M2roe3FdlmRLeXs9cOH00KrccLrazkDvlYGSg_N9upo,410
-omnigenome/auto/auto_bench/auto_bench.py,sha256=BdCzS-KF_Eos22TGlNdTtLFXlNt0Hd6yYpfvb1TawjE,20997
-omnigenome/auto/auto_bench/auto_bench_cli.py,sha256=B3iJiTKuAc6dkEu2h9ApGFpJJwS0I2zwxXtZ6bWfarM,7608
-omnigenome/auto/auto_bench/auto_bench_config.py,sha256=2mw6VFcWCSR6dJVAGDejlwmK_VmAZgqMvYlqznl5Ck8,7723
-omnigenome/auto/auto_bench/config_check.py,sha256=5RYOOOC-CqvU-46-3uSK6vspLtPSVxdZsFMnNa-3rj0,1086
-omnigenome/auto/auto_train/__init__.py,sha256=JClK_1E1_YVAy2rNzQewYzYxvLanqm71-swdXAx8uZs,447
-omnigenome/auto/auto_train/auto_train.py,sha256=af5ca6SGyxn8apIxwU0vKPoAC0mKDP2CFqN9w1N_a2M,17506
-omnigenome/auto/auto_train/auto_train_cli.py,sha256=vSjQ1t3hfe4HEvIWVB7V6cmYf-47_obKZGHWrk_LWDc,7298
-omnigenome/auto/bench_hub/__init__.py,sha256=JryopzWvewnBtN_EZKNEDKvIsc6x0wCsntNFyCGhpmU,395
-omnigenome/auto/bench_hub/bench_hub.py,sha256=eTxcVrKQdb-tdgt4CZDmD7zsUxMdw6McOeDAuaDiQZc,740
-omnigenome/cli/__init__.py,sha256=sLM_HMGX4lK8LKvIsIBmQUAbZKih6bfYoHmqQFKfGBM,445
-omnigenome/cli/omnigenome_cli.py,sha256=fvpxWqowa7IFSU097EXrhChYcMN4yekGS9UA5nsbpz4,4386
-omnigenome/cli/commands/__init__.py,sha256=nnD9NFBNHGYhUt7nI5y4iuylIn2JOdvn6iEhlDJjhes,435
-omnigenome/cli/commands/base.py,sha256=KU_lhhZpjCMRAK0LZz_txV3R9ZfzCAo_k3u44cBztMk,2858
-omnigenome/cli/commands/bench/__init__.py,sha256=dAVl47_QKktT5DpHjv0VMY18vlmxkNMSXbfIWZ1l69k,440
-omnigenome/cli/commands/bench/bench_cli.py,sha256=N4mkhg2MRo3beAWmUDJe2NsB5vL3hquxs8fBN-c9Z-A,7523
-omnigenome/cli/commands/rna/__init__.py,sha256=rUyJljDj9JGcn83iyRSB2qC2BAXkgsbb24SWsimOFVE,434
-omnigenome/cli/commands/rna/rna_design.py,sha256=Vb9jhh4Aj8RlUsBSHirQY5supkmMYzaXNIbs32Jjsyo,6408
-omnigenome/src/__init__.py,sha256=MtntWq77RoL1K6iUQQlVcwgfDZIrykvjcpP4O4_7jgM,412
-omnigenome/src/abc/__init__.py,sha256=4E5Ba2Fwf9VlxmjwZ50kku-YPGU00JZPgAcNuesaG3g,424
-omnigenome/src/abc/abstract_dataset.py,sha256=YEHMwqohAPozmmbVemzA7CoA2hKyvrkW1wIB1msEQqA,24254
-omnigenome/src/abc/abstract_metric.py,sha256=nRISSO93i1L9Q-TDL0ZnhtKAXgLRKVlzL6oKHknFU-4,4326
-omnigenome/src/abc/abstract_model.py,sha256=t1XqLqs7VLJ4I9Tzu0chWYyFr3_ElaHk5vzQK5JGa7E,28639
-omnigenome/src/abc/abstract_tokenizer.py,sha256=5z5dhL2qSswTa5W7Gdn7u8q7sC8cBfv2JuJf3G-F8Gw,9888
-omnigenome/src/dataset/__init__.py,sha256=2Lz3xfGaiP5hl2Pf1eW0eenpzrDk3zDyyi63CXbC4hE,628
-omnigenome/src/dataset/omni_dataset.py,sha256=dFsDraF3x9N-JIsxbAyPiDhY74BIdWn2pg0A_bV8iuc,17609
-omnigenome/src/lora/__init__.py,sha256=EncdIW5iV7LWh8JndQqoXIDTZgHWAIGaeNUEyMa1SlU,457
-omnigenome/src/lora/lora_model.py,sha256=HX_CW_ZlwHjGWLY-wcL01_6X6fIQKS7QxDZrmdmll44,10101
-omnigenome/src/metric/__init__.py,sha256=lZr6uiL44G8ujmCz98wTH_anHRv9QUdO3tUiBHriZXA,543
-omnigenome/src/metric/classification_metric.py,sha256=9AEBqtwga6cojFcS1lP_i-LGkbmn3QLAD7JGFz2Z6bY,8030
-omnigenome/src/metric/metric.py,sha256=lAn7brhugv-gOEnE3SatnhczjWDgV8JTLfWxJU-regc,7724
-omnigenome/src/metric/ranking_metric.py,sha256=iQZd7DIHIiNqykQJPNPVVjmGlrV-6nSEeKIh5o2XUt4,5670
-omnigenome/src/metric/regression_metric.py,sha256=hfUPD6TnrfrQ_wpW5Lyk6t6xdHyMWhJ0rthsk3MXJkw,7585
-omnigenome/src/misc/__init__.py,sha256=Dpa-uCQdwKVKkprqy26Np71mRobcWglCjgtITjU6yw0,63
-omnigenome/src/misc/utils.py,sha256=WHoXcRpwliV1-9YO5ZfttvJNzUGib77uHS6oNVUnVjo,17108
-omnigenome/src/model/__init__.py,sha256=vu1vJVYp8FR9BgF7X2msKkwMfa6jbzsfAsUHduTB21w,621
-omnigenome/src/model/module_utils.py,sha256=dmiORa835Jg6cGS24h8e1ZrwfB9_tGiz30IbwUagdaw,9158
-omnigenome/src/model/augmentation/__init__.py,sha256=Q1uTFVJjlDV_nJZ9YP44gq6LSolz9b8_6gJ0Iz0RtAc,395
-omnigenome/src/model/augmentation/model.py,sha256=rA-NZGzydqvpZuHy4fq4xVJcFA97rjb6NcoV-Leeaqg,8174
-omnigenome/src/model/classification/__init__.py,sha256=T1pW6HLdjUywf4-mN9d4ERRWPiZsGpDqsbalFPtyVCc,399
-omnigenome/src/model/classification/model.py,sha256=wAets3HAF_T66LbBmdoEutKMYAwaIvWvMffDgGhnUbU,25022
-omnigenome/src/model/embedding/__init__.py,sha256=Q6UqgzCEOujfAvYKcHnYhCa6kMIzFFFIa_5Ut8iHQgg,394
-omnigenome/src/model/embedding/model.py,sha256=JJIjQTy50U8fVzeeXbgvUdJv_W59dF-f0gZD6qB8ha8,10140
-omnigenome/src/model/mlm/__init__.py,sha256=PqKckKLXhvkJr8dLkSpKTwFYsclGv51nhXBKjI55oYw,406
-omnigenome/src/model/mlm/model.py,sha256=rFYjn_QYFcREqHzSGO8GsbdUGGMzjOkAcooa-P9SaQ8,6435
-omnigenome/src/model/regression/__init__.py,sha256=9_PEGxaiJrwPXvzOf6jGx6Wl1x4UQlOsCGUwNYPUDbc,395
-omnigenome/src/model/regression/model.py,sha256=lXLg1Z0B7jzXHRkt8F1sP3KAitfSIAvhe8EAOFiUMFI,27591
-omnigenome/src/model/regression/resnet.py,sha256=Xvpssyo3LmmPMJYHwdqKHU0x9F39dllkKY94aI9mHO0,16861
-omnigenome/src/model/rna_design/__init__.py,sha256=cqC0Ap954sBDOe3RI8nVlYahx_gIg0nqwoBg8CRQ62E,395
-omnigenome/src/model/rna_design/model.py,sha256=kDDM0BHpoFN9nlGxNWrjTjsFWn7E6CPYjfAfkJyLxGY,16939
-omnigenome/src/model/seq2seq/__init__.py,sha256=38YtCYxAQRytOsW98G8cqgCXB6mmzrSGPKs1DbwzZ40,405
-omnigenome/src/model/seq2seq/model.py,sha256=UC7j691BSbr6He3kCYwO_93V4RLWLt3fQOCQnnwJwgQ,1695
-omnigenome/src/tokenizer/__init__.py,sha256=zYUgX-FJ-fw0GNJuuW8ovo9kflDmGDd8Z0F3AMDFXF4,556
-omnigenome/src/tokenizer/bpe_tokenizer.py,sha256=_g1x3jJ3wzGgDu3_y2pW-KghYllgbiqoJfjMUF3sNTE,7646
-omnigenome/src/tokenizer/kmers_tokenizer.py,sha256=KpPA5_MSP-Wvbp2aeRGD67al6xS1unhbreIQOio-1U4,9155
-omnigenome/src/tokenizer/single_nucleotide_tokenizer.py,sha256=alwWeXuvGvzGflsyzm_V0AelAbdfMs79FGSo4pL5roo,9710
-omnigenome/src/trainer/__init__.py,sha256=adib_MJlED-EXXkNCt9MhV1U9mPPoOrLReA-dxhgdT0,453
-omnigenome/src/trainer/accelerate_trainer.py,sha256=u0gU1MRLjEcj3zPAkDztjxU1aNWDA_XTl6k_Wk3esIs,28377
-omnigenome/src/trainer/hf_trainer.py,sha256=eRw0v21lBZW1gTi-5zmagLuR0A1X7OOb0bZlmzROcMI,2591
-omnigenome/src/trainer/trainer.py,sha256=wyHJcb8IhrIrSW9dE7uK5MU_EL0_LRHJBP65z7RW9gA,21333
-omnigenome/utility/__init__.py,sha256=UvgBH3hsW2JoXfGG6dHJeQLHVF7rXdzK39wGCoJOEKI,101
-omnigenome/utility/ensemble.py,sha256=z9opb_9H213OeeNn4VJbKQfXVK_Kzig3mNDIjviG1oQ,13141
-omnigenome/utility/hub_utils.py,sha256=iYxt1iAKW2OTsQ0rYxMixRiarwIxjYzCQtaDyijFnWw,20269
-omnigenome/utility/dataset_hub/__init__.py,sha256=86rjBJixstpSiTbtprf0RnaC1V5lykh2Bs2cnZG9j3s,433
-omnigenome/utility/dataset_hub/dataset_hub.py,sha256=aTPfk5Owoa5OtSGbFPM34WmZ3Oy9x8ZrILQgF6VXI3g,6960
-omnigenome/utility/model_hub/__init__.py,sha256=Y5AcnzrtX9zPVbsAZy-6kLPJef-eddcPtZRAe8N7BOE,391
-omnigenome/utility/model_hub/model_hub.py,sha256=0K0XOQvPLiscHt0LRphF7Gja6MXHDgBARbdQzr1G8mI,8574
-omnigenome/utility/pipeline_hub/__init__.py,sha256=lz9kdc_arVwqDENvt_dN0h8fAhBeTRyG6LQRQrQAjGE,394
-omnigenome/utility/pipeline_hub/pipeline.py,sha256=Ft56dolrLJWSLBj7jba8Fka3yCwtBso2Vx_Nt17ZB7A,19756
-omnigenome/utility/pipeline_hub/pipeline_hub.py,sha256=bt2QLg4cld8QT2fbMZXJKL0vE3ReuNWHsqhLgiwc6HI,5268
-omnigenome-0.3.1a0.dist-info/licenses/LICENSE,sha256=oQoefBV6siHctF0ET-OO3EaSZgtqGtf-wdIAmokS8iY,11560
-omnigenome-0.3.1a0.dist-info/METADATA,sha256=LA0FxjRSjsEhsXvOV3noKYAis1t4_JY-wqTDjV9Fszo,10315
-omnigenome-0.3.1a0.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
-omnigenome-0.3.1a0.dist-info/entry_points.txt,sha256=uu40UgMPxY65ASdRbrhkwH94r7CIYgyG_iDBmqFQbD8,84
-omnigenome-0.3.1a0.dist-info/top_level.txt,sha256=LVFxm_WPaxjj9KnAqdW94W4D4lbOk30gdsaKlJiSzTo,11
-omnigenome-0.3.1a0.dist-info/RECORD,,