atomgen.data.utils module# Utilities for data processing and evaluation. compute_metrics_lba(eval_pred)[source]# Compute RMSE for the LBA task. Return type: Dict[str, Any] compute_metrics_lep(eval_pred)[source]# Compute AUROC for the LEP task. Return type: Dict[str, Any] compute_metrics_msp(eval_pred)[source]# Compute AUROC for the MSP task. Return type: Dict[str, Any] compute_metrics_ppi(eval_pred)[source]# Compute AUROC for the PIP task. Return type: Dict[str, Any] compute_metrics_psr(eval_pred)[source]# Compute global spearman correlation for the PSR task. Return type: Dict[str, Any] compute_metrics_res(eval_pred)[source]# Compute accuracy for the RES task. Return type: Dict[str, Any] compute_metrics_rsr(eval_pred)[source]# Compute global spearman correlation for the RSR task. Return type: Dict[str, Any] compute_metrics_smp(eval_pred)[source]# Compute MAE for 20 regression labels for the SMP task. Return type: Dict[str, Any]