Error in predict_risk(test_data)

[XPeriment2]

After following the notebook: model.train(train_data, n_epochs=n_epochs, logger=logger, update_fn=update_fn) i use model.get_concordance_index(**test_data) and i get a ci. Howver: predictions =model.predict_risk(test_data) prodcues an error of excetpion

[hdplsa]

When you call model.predict_risk you should only pass the x matrix. I think you should be able to fix your problem by calling instead:

predictions = model.predict_risk(test_data["x"])