Finish co-occurrence model training function.