crbn4/Building
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified: __pycache__/evaluation_studio.cpython-314.pyc

Browse Source 
modified:   evaluation_studio.py
	modified:   scripts/dataset.py
	new file:   scripts/inference_utils.py
	modified:   scripts/predict_single.py
	modified:   scripts_2/__pycache__/task3_identify.cpython-314.pyc
	modified:   scripts_2/task3_identify.py
This commit is contained in:
CrbnsCat10n
2026-05-07 15:56:31 +08:00
parent d2dde4f557
commit 3d40e88e35
7 changed files with 285 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/predict_single.py
View File

@@ -10,9 +10,11 @@ import numpy as np
try:
from .config import CORE_FEATURE_NAMES, evaluation_dir, make_experiment_config
from .dataset import build_record_from_file
from .inference_utils import predict_task1_tr
except ImportError:
from config import CORE_FEATURE_NAMES, evaluation_dir, make_experiment_config
from dataset import build_record_from_file
from inference_utils import predict_task1_tr
def parse_args() -> argparse.Namespace:
@@ -87,7 +89,14 @@ def main() -> None:
file_path = Path(args.file).resolve()
record = build_record_from_file(file_path, config.data)
pred_tr = max(float(model.predict([record.features.tolist()])[0]), config.data.normalization_eps)
pred_tr, tr_source = predict_task1_tr(
model,
feature_vector=record.features,
frequency_hz=float(record.frequency_hz),
normalization_eps=float(config.data.normalization_eps),
project_root=config.data.project_root,
prefer_curve=True,
)
pred_rms = pred_tr * record.x_rms
out_dir = evaluation_dir(config)
@@ -99,6 +108,7 @@ def main() -> None:
for feature_name, feature_value in zip(CORE_FEATURE_NAMES, record.features.tolist()):
print(f"{feature_name}: {feature_value:.6f}")
print(f"Predicted TR: {pred_tr:.6f}")
print(f"TR Source: {tr_source}")
print(f"Predicted RMS: {pred_rms:.6f}")
print(f"True RMS: {record.y_rms:.6f}")
print(f"Relative RMS Error (%): {abs(pred_rms - record.y_rms) / max(abs(record.y_rms), 1e-12) * 100.0:.4f}")