Examples with ModelSaveContext com.tencent.angel.model.ModelSaveContext used on opensource projects

Example 6 with ModelSaveContext

use of com.tencent.angel.model.ModelSaveContext in project angel by Tencent.

the class AngelClient method saveMatrices.

/**
 * Save matrices to files.
 *
 * @param matrixNames need save matrix name list
 */
public void saveMatrices(List<String> matrixNames) {
    ModelSaveContext saveContext = new ModelSaveContext();
    saveContext.setSavePath(conf.get(AngelConf.ANGEL_JOB_OUTPUT_PATH));
    for (String name : matrixNames) {
        saveContext.addMatrix(new MatrixSaveContext(name));
    }
    save(saveContext);
}

Example 7 with ModelSaveContext

use of com.tencent.angel.model.ModelSaveContext in project angel by Tencent.

the class ProtobufUtil method convert.

public static ModelSaveContext convert(ModelSaveContextProto saveContext) {
    ModelSaveContext context = new ModelSaveContext(saveContext.getSavePath());
    context.setCheckpoint(saveContext.getCheckpint());
    List<MatrixSaveContextProto> matricesContext = saveContext.getMatrixContextesList();
    int size = matricesContext.size();
    for (int i = 0; i < size; i++) {
        context.addMatrix(convert(matricesContext.get(i)));
    }
    return context;
}