Skip to content

Benchmark and Model Evaluation Metrics #11

Benchmark and Model Evaluation Metrics

Benchmark and Model Evaluation Metrics #11

name: Check Training Parameters
on:
workflow_dispatch:
pull_request:
branches: [ main ]
paths:
- 'XPointMLTest.py' # Only trigger when this specific file changes
jobs:
check-training-params:
runs-on: ubuntu-latest
steps:
- name: Checkout PR code
uses: actions/checkout@v3
with:
ref: ${{ github.event.pull_request.head.sha }}
path: pr-code
fetch-depth: 1
- name: Checkout main branch
uses: actions/checkout@v3
with:
ref: main
path: main-code
- name: Set up Python
uses: actions/setup-python@v4
with:
python-version: '3.10'
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install pytest
- name: Run parameter check
run: |
python - <<EOF
import re
import sys
# Training parameter patterns to check
params_to_check = {
'epochs': r'--epochs\', type=int, default=(\d+),',
'trainFrameFirst': r'--trainFrameFirst\', type=int, default=(\d+),',
'trainFrameLast': r'--trainFrameLast\', type=int, default=(\d+),',
'validationFrameFirst': r'--validationFrameFirst\', type=int, default=(\d+),',
'validationFrameLast': r'--validationFrameLast\', type=int, default=(\d+),',
}
# Files to check
files_to_check = ['main-code/XPointMLTest.py', 'pr-code/XPointMLTest.py']
main_params = {}
pr_params = {}
# Extract parameters from main branch
with open('main-code/XPointMLTest.py', 'r') as f:
content = f.read()
for param, pattern in params_to_check.items():
match = re.search(pattern, content)
if match:
main_params[param] = int(match.group(1))
else:
print(f"Warning: Could not find parameter '{param}' in main branch code")
# Extract parameters from PR
with open('pr-code/XPointMLTest.py', 'r') as f:
content = f.read()
for param, pattern in params_to_check.items():
match = re.search(pattern, content)
if match:
pr_params[param] = int(match.group(1))
else:
print(f"Warning: Could not find parameter '{param}' in PR code")
# Compare parameters
mismatch = False
for param in params_to_check.keys():
if param in main_params and param in pr_params:
if main_params[param] != pr_params[param]:
print(f"❌ Parameter '{param}' has changed: {main_params[param]} -> {pr_params[param]}")
mismatch = True
else:
print(f"✅ Parameter '{param}' unchanged: {main_params[param]}")
else:
print(f"⚠️ Could not compare '{param}' - missing from one or both branches")
# Summary
print("\n=== Parameter Check Summary ===")
if mismatch:
print("❌ Training parameters have been modified!")
print("Detected changes to training configuration parameters.")
print("Please verify these changes are intentional and approved.")
sys.exit(1)
else:
print("✅ All training parameters match the main branch!")
sys.exit(0)
EOF