vtkPSciVizKMeans.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkPSciVizKMeans_h
#define vtkPSciVizKMeans_h

#include "vtkPVVTKExtensionsFiltersStatisticsModule.h" //needed for exports
#include "vtkSciVizStatistics.h"

class VTKPVVTKEXTENSIONSFILTERSSTATISTICS_EXPORT vtkPSciVizKMeans : public vtkSciVizStatistics
{
public:
  static vtkPSciVizKMeans* New();
  vtkTypeMacro(vtkPSciVizKMeans, vtkSciVizStatistics);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetMacro(K, int);
  vtkGetMacro(K, int);


  vtkSetMacro(MaxNumIterations, int);
  vtkGetMacro(MaxNumIterations, int);


  vtkSetMacro(Tolerance, double);
  vtkGetMacro(Tolerance, double);

protected:
  vtkPSciVizKMeans();
  ~vtkPSciVizKMeans() override;

  int LearnAndDerive(vtkMultiBlockDataSet* model, vtkTable* inData) override;
  int AssessData(
    vtkTable* observations, vtkDataObject* dataset, vtkMultiBlockDataSet* model) override;

  int K;
  int MaxNumIterations;
  double Tolerance;

private:
  vtkPSciVizKMeans(const vtkPSciVizKMeans&) = delete;
  void operator=(const vtkPSciVizKMeans&) = delete;
};

#endif // vtkPSciVizKMeans_h