search

ucl-pond / pySuStaIn

Subtype and Stage Inference (SuStaIn) algorithm with an example using simulated data.
MIT License
122 stars 63 forks source link

Cross-validation for selecting optimal number of clusters #2

Closed noxtoby closed 4 years ago

noxtoby commented 5 years ago

We should code up a script that implements 10-fold CV for selecting the optimal number of subtype (see the SuStaIn paper). For each fold:

  1. Calculate model likelihood for test and training sets;
  2. Output/Save this, and optionally also plot.
ayoung11 commented 4 years ago

Leon added this functionality so I'm closing this comment