Yiqi Deng
Yiqi Deng
```python model = stg.STG(task_type='cox',input_dim=train_data_K['X'].shape[1], output_dim=1, hidden_dims=[60, 20, 2],activation='selu', optimizer='Adam', learning_rate=1e-3, batch_size=train_data_K['X'].shape[0], feature_selection=feature_selection, sigma=0.5, lam=0.001, random_state=123, device=device) model.fit(train_data_K['X'], {'E': train_data_K['E'],'T': train_data_K['T']}, valid_X=val_data_K['X'], valid_y= {'E': val_data_K['E'], 'T': val_data_K['T']}, nr_epochs=460, print_interval=100) ## get...