Files
vinatools/scripts/example_api_usage.py
2025-10-15 20:14:12 +08:00

30 lines
1.1 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Example usage of the analyze_qed_mw_distribution API
"""
from analyze_qed_mw_distribution import main_api
print("Running analysis examples...")
# Example 1: Basic usage
print("\nExample 1: Basic usage")
main_api(['qed_values_fgbar.csv', 'qed_values_trpe.csv'], ['fgbar', 'trpe'])
# Example 2: With custom reference scores
print("\nExample 2: With custom reference scores")
main_api(['qed_values_fgbar.csv', 'qed_values_trpe.csv'], ['fgbar', 'trpe'],
reference_scores={'fgbar': {'9NY': -5.268}, 'trpe': {'0GA': -6.531}})
# Example 3: With specific conformation rank
print("\nExample 3: With specific conformation rank")
main_api(['qed_values_fgbar.csv', 'qed_values_trpe.csv'], ['fgbar', 'trpe'], rank=0)
# Example 4: With both custom reference scores and specific conformation rank
print("\nExample 4: With both custom reference scores and specific conformation rank")
main_api(['qed_values_fgbar.csv', 'qed_values_trpe.csv'], ['fgbar', 'trpe'],
reference_scores={'fgbar': {'9NY': -5.268}, 'trpe': {'0GA': -6.531}}, rank=0)
print("\nAnalysis complete! Check the generated PNG files.")