havakv / pycox

Survival analysis with PyTorch
BSD 2-Clause "Simplified" License
780 stars 180 forks source link

AssertionError: assert durations.shape[0] == surv.shape[1] == surv_idx.shape[0] == events.shape[0] #159

Open nanhui2000 opened 1 year ago

nanhui2000 commented 1 year ago

[W 2023-04-23 10:28:38,810] Trial 2 failed with parameters: {'epochs': 1782, 'batch_norm': False, 'dropout': 0.3729027863986679, 'batch_size': 128, 'alpha': 0.20950098112214738, 'sigma': 0.021532851493345717, 'num_layers': 2, 'learning_rate': 0.0004455921289602272, 'optimizer_type': 'AdamWR', 'num_nodes_1': 56, 'num_nodes_2': 20} because of the following error: AssertionError(). Traceback (most recent call last): File "/home/anaconda3/envs/py3.7/lib/python3.7/site-packages/optuna/study/_optimize.py", line 200, in _run_trial value_or_values = func(trial) File "/tmp/ipykernel_25107/2073144205.py", line 132, in objective c_index = ev.concordance_td('antolini') File "/home/anaconda3/envs/py3.7/lib/python3.7/site-packages/pycox/evaluation/eval_surv.py", line 191, in concordance_td self._duration_idx(), method) File "/home/anaconda3/envs/py3.7/lib/python3.7/site-packages/pycox/evaluation/concordance.py", line 93, in concordance_td assert durations.shape[0] == surv.shape[1] == surv_idx.shape[0] == events.shape[0] AssertionError [W 2023-04-23 10:28:38,812] Trial 2 failed with value None. [W 2023-04-23 10:28:46,183] Trial 1 failed with parameters: {'epochs': 1947, 'batch_norm': False, 'dropout': 0.29937082041280605, 'batch_size': 256, 'alpha': 0.3882041406677713, 'sigma': 0.7061223972910085, 'num_layers': 1, 'learning_rate': 0.001519199989009398, 'optimizer_type': 'AdamW', 'num_nodes_1': 54} because of the following error: AssertionError(). Traceback (most recent call last): File "/home/anaconda3/envs/py3.7/lib/python3.7/site-packages/optuna/study/_optimize.py", line 200, in _run_trial value_or_values = func(trial) File "/tmp/ipykernel_25107/2073144205.py", line 132, in objective c_index = ev.concordance_td('antolini') File "/home/anaconda3/envs/py3.7/lib/python3.7/site-packages/pycox/evaluation/eval_surv.py", line 191, in concordance_td self._duration_idx(), method) File "/home/anaconda3/envs/py3.7/lib/python3.7/site-packages/pycox/evaluation/concordance.py", line 93, in concordance_td assert durations.shape[0] == surv.shape[1] == surv_idx.shape[0] == events.shape[0] AssertionError