Lines Matching refs:regressor

66   PetscRegressor regressor;  in PetscRegressorCreate()  local
73 …PetscCall(PetscHeaderCreate(regressor, PETSCREGRESSOR_CLASSID, "PetscRegressor", "Regressor", "Pet… in PetscRegressorCreate()
75 regressor->setupcalled = PETSC_FALSE; in PetscRegressorCreate()
76 regressor->fitcalled = PETSC_FALSE; in PetscRegressorCreate()
77 regressor->data = NULL; in PetscRegressorCreate()
78 regressor->training = NULL; in PetscRegressorCreate()
79 regressor->target = NULL; in PetscRegressorCreate()
80 …PetscObjectParameterSetDefault(regressor, regularizer_weight, 1.0); // Default to regularizer weig… in PetscRegressorCreate()
82 *newregressor = regressor; in PetscRegressorCreate()
110 PetscErrorCode PetscRegressorView(PetscRegressor regressor, PetscViewer viewer) in PetscRegressorView() argument
116 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorView()
117 if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)regressor)->comm, &viewer)); in PetscRegressorView()
119 PetscCheckSameComm(regressor, 1, viewer, 2); in PetscRegressorView()
124 PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)regressor, viewer)); in PetscRegressorView()
127 PetscTryTypeMethod(regressor, view, viewer); in PetscRegressorView()
128 if (regressor->tao) PetscCall(TaoView(regressor->tao, viewer)); in PetscRegressorView()
131 PetscCall(PetscRegressorGetType(regressor, &type)); in PetscRegressorView()
178 PetscErrorCode PetscRegressorSetFromOptions(PetscRegressor regressor) in PetscRegressorSetFromOptions() argument
185 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorSetFromOptions()
186 if (((PetscObject)regressor)->type_name) default_type = ((PetscObject)regressor)->type_name; in PetscRegressorSetFromOptions()
187 PetscObjectOptionsBegin((PetscObject)regressor); in PetscRegressorSetFromOptions()
191 PetscCall(PetscRegressorSetType(regressor, type)); in PetscRegressorSetFromOptions()
192 } else if (!((PetscObject)regressor)->type_name) { in PetscRegressorSetFromOptions()
193 PetscCall(PetscRegressorSetType(regressor, default_type)); in PetscRegressorSetFromOptions()
195 …the regularizer", "PetscRegressorSetRegularizerWeight", regressor->regularizer_weight, &regressor-… in PetscRegressorSetFromOptions()
196 if (flg) PetscCall(PetscRegressorSetRegularizerWeight(regressor, regressor->regularizer_weight)); in PetscRegressorSetFromOptions()
198 PetscTryTypeMethod(regressor, setfromoptions, PetscOptionsObject); in PetscRegressorSetFromOptions()
223 PetscErrorCode PetscRegressorSetUp(PetscRegressor regressor) in PetscRegressorSetUp() argument
226 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorSetUp()
227 if (regressor->setupcalled) PetscFunctionReturn(PETSC_SUCCESS); in PetscRegressorSetUp()
228 PetscCall(PetscLogEventBegin(PetscRegressor_SetUp, regressor, 0, 0, 0)); in PetscRegressorSetUp()
230 PetscTryTypeMethod(regressor, setup); in PetscRegressorSetUp()
231 regressor->setupcalled = PETSC_TRUE; in PetscRegressorSetUp()
232 PetscCall(PetscLogEventEnd(PetscRegressor_SetUp, regressor, 0, 0, 0)); in PetscRegressorSetUp()
252 PetscErrorCode PetscRegressorFit(PetscRegressor regressor, Mat X, Vec y) in PetscRegressorFit() argument
255 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorFit()
261 PetscCall(MatDestroy(&regressor->training)); in PetscRegressorFit()
262 regressor->training = X; in PetscRegressorFit()
266 PetscCall(VecDestroy(&regressor->target)); in PetscRegressorFit()
267 regressor->target = y; in PetscRegressorFit()
269 PetscCall(PetscRegressorSetUp(regressor)); in PetscRegressorFit()
271 PetscCall(PetscLogEventBegin(PetscRegressor_Fit, regressor, X, y, 0)); in PetscRegressorFit()
272 PetscUseTypeMethod(regressor, fit); in PetscRegressorFit()
273 PetscCall(PetscLogEventEnd(PetscRegressor_Fit, regressor, X, y, 0)); in PetscRegressorFit()
275 PetscCall(PetscRegressorViewFromOptions(regressor, NULL, "-regressor_view")); in PetscRegressorFit()
276 regressor->fitcalled = PETSC_TRUE; in PetscRegressorFit()
296 PetscErrorCode PetscRegressorPredict(PetscRegressor regressor, Mat X, Vec y) in PetscRegressorPredict() argument
299 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorPredict()
302 …PetscCheck(regressor->fitcalled == PETSC_TRUE, ((PetscObject)regressor)->comm, PETSC_ERR_ARG_WRONG… in PetscRegressorPredict()
303 PetscCall(PetscLogEventBegin(PetscRegressor_Predict, regressor, X, y, 0)); in PetscRegressorPredict()
304 PetscTryTypeMethod(regressor, predict, X, y); in PetscRegressorPredict()
305 PetscCall(PetscLogEventEnd(PetscRegressor_Predict, regressor, X, y, 0)); in PetscRegressorPredict()
321 PetscErrorCode PetscRegressorReset(PetscRegressor regressor) in PetscRegressorReset() argument
324 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorReset()
325 if (regressor->ops->reset) PetscTryTypeMethod(regressor, reset); in PetscRegressorReset()
326 PetscCall(MatDestroy(&regressor->training)); in PetscRegressorReset()
327 PetscCall(VecDestroy(&regressor->target)); in PetscRegressorReset()
328 PetscCall(TaoDestroy(&regressor->tao)); in PetscRegressorReset()
329 regressor->setupcalled = PETSC_FALSE; in PetscRegressorReset()
330 regressor->fitcalled = PETSC_FALSE; in PetscRegressorReset()
346 PetscErrorCode PetscRegressorDestroy(PetscRegressor *regressor) in PetscRegressorDestroy() argument
349 if (!*regressor) PetscFunctionReturn(PETSC_SUCCESS); in PetscRegressorDestroy()
350 PetscValidHeaderSpecific(*regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorDestroy()
351 if (--((PetscObject)*regressor)->refct > 0) { in PetscRegressorDestroy()
352 *regressor = NULL; in PetscRegressorDestroy()
356 PetscCall(PetscRegressorReset(*regressor)); in PetscRegressorDestroy()
357 PetscTryTypeMethod(*regressor, destroy); in PetscRegressorDestroy()
359 PetscCall(PetscHeaderDestroy(regressor)); in PetscRegressorDestroy()
390 PetscErrorCode PetscRegressorSetType(PetscRegressor regressor, PetscRegressorType type) in PetscRegressorSetType() argument
396 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorSetType()
399 PetscCall(PetscObjectTypeCompare((PetscObject)regressor, type, &match)); in PetscRegressorSetType()
403 …PetscCheck(r, PetscObjectComm((PetscObject)regressor), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find… in PetscRegressorSetType()
406 PetscTryTypeMethod(regressor, destroy); in PetscRegressorSetType()
407 PetscCall(TaoDestroy(&regressor->tao)); in PetscRegressorSetType()
408 regressor->ops->setup = NULL; in PetscRegressorSetType()
409 regressor->ops->setfromoptions = NULL; in PetscRegressorSetType()
410 regressor->ops->settraining = NULL; in PetscRegressorSetType()
411 regressor->ops->fit = NULL; in PetscRegressorSetType()
412 regressor->ops->predict = NULL; in PetscRegressorSetType()
413 regressor->ops->destroy = NULL; in PetscRegressorSetType()
414 regressor->ops->reset = NULL; in PetscRegressorSetType()
415 regressor->ops->view = NULL; in PetscRegressorSetType()
418 regressor->setupcalled = PETSC_FALSE; in PetscRegressorSetType()
419 PetscCall((*r)(regressor)); in PetscRegressorSetType()
420 PetscCall(PetscObjectChangeTypeName((PetscObject)regressor, type)); in PetscRegressorSetType()
439 PetscErrorCode PetscRegressorGetType(PetscRegressor regressor, PetscRegressorType *type) in PetscRegressorGetType() argument
442 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorGetType()
444 *type = ((PetscObject)regressor)->type_name; in PetscRegressorGetType()
464 PetscErrorCode PetscRegressorSetRegularizerWeight(PetscRegressor regressor, PetscReal weight) in PetscRegressorSetRegularizerWeight() argument
467 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorSetRegularizerWeight()
468 PetscValidLogicalCollectiveReal(regressor, weight, 2); in PetscRegressorSetRegularizerWeight()
469 regressor->regularizer_weight = weight; in PetscRegressorSetRegularizerWeight()
497 PetscErrorCode PetscRegressorGetTao(PetscRegressor regressor, Tao *tao) in PetscRegressorGetTao() argument
500 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorGetTao()
503 if (!regressor->tao) { in PetscRegressorGetTao()
504 PetscCall(TaoCreate(PetscObjectComm((PetscObject)regressor), &regressor->tao)); in PetscRegressorGetTao()
505 PetscCall(PetscObjectIncrementTabLevel((PetscObject)regressor->tao, (PetscObject)regressor, 1)); in PetscRegressorGetTao()
506 … PetscCall(PetscObjectSetOptions((PetscObject)regressor->tao, ((PetscObject)regressor)->options)); in PetscRegressorGetTao()
508 *tao = regressor->tao; in PetscRegressorGetTao()
543 PetscErrorCode PetscRegressorSetOptionsPrefix(PetscRegressor regressor, const char p[]) in PetscRegressorSetOptionsPrefix() argument
546 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorSetOptionsPrefix()
547 PetscCall(PetscObjectSetOptionsPrefix((PetscObject)regressor, p)); in PetscRegressorSetOptionsPrefix()
568 PetscErrorCode PetscRegressorAppendOptionsPrefix(PetscRegressor regressor, const char p[]) in PetscRegressorAppendOptionsPrefix() argument
571 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorAppendOptionsPrefix()
572 PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)regressor, p)); in PetscRegressorAppendOptionsPrefix()
595 PetscErrorCode PetscRegressorGetOptionsPrefix(PetscRegressor regressor, const char *p[]) in PetscRegressorGetOptionsPrefix() argument
598 PetscValidHeaderSpecific(regressor, PETSCREGRESSOR_CLASSID, 1); in PetscRegressorGetOptionsPrefix()
599 PetscCall(PetscObjectGetOptionsPrefix((PetscObject)regressor, p)); in PetscRegressorGetOptionsPrefix()