BSB_Sim/experiments/03_feature_ablation.py
2026-10-19 08:11:33 +02:00

35 lines
1.0 KiB
Python

"""Compare DTW with optional pressure and contact-area channels."""
import argparse
try:
from experiments.common import evaluate_directory
except ModuleNotFoundError:
from common import evaluate_directory
def main() -> None:
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--data-dir", default="data/raw/mobisig")
parser.add_argument("--enrollment-count", type=int, default=3)
args = parser.parse_args()
configurations = [
("position and motion", False, False),
("plus pressure", True, False),
("plus contact area", False, True),
("pressure and area", True, True),
]
for name, pressure, area in configurations:
metrics = evaluate_directory(
args.data_dir,
enrollment_count=args.enrollment_count,
include_pressure=pressure,
include_area=area,
mobisig=True,
)
print(f"{name}: EER={metrics['eer']:.4f}")
if __name__ == "__main__":
main()