in sourcecode/scoring/reputation_matrix_factorization/diligence_model.py [0:0]
def _prepare_diligence_init_state(noteInitState, raterInitState):
if noteInitState is not None:
noteInitState = noteInitState[
[c.noteIdKey] + [col for col in noteInitState.columns if "lowDiligence" in col]
]
noteInitState.columns = [
col.replace("lowDiligence", "internal") for col in noteInitState.columns
]
if raterInitState is not None:
raterInitState = raterInitState[
[c.raterParticipantIdKey] + [col for col in raterInitState.columns if "lowDiligence" in col]
]
raterInitState.columns = [
col.replace("lowDiligence", "internal") for col in raterInitState.columns
]
return noteInitState, raterInitState