Differential D22614 Diff 62197 src/backend/generalTest/CorrelationCoefficient.h

Changeset View

Standalone View

src/backend/generalTest/CorrelationCoefficient.h

This file was added.

1		/***************************************************************************
2		File : CorrelationCoefficient.h
3		Project : LabPlot
4		Description : Finding Correlation Coefficient on data provided
5		--------------------------------------------------------------------
6		Copyright : (C) 2019 Devanshu Agarwal(agarwaldevanshu8@gmail.com)
7
8		***************************************************************************/
9
10		/***************************************************************************
11		* *
12		* This program is free software; you can redistribute it and/or modify *
13		* it under the terms of the GNU General Public License as published by *
14		* the Free Software Foundation; either version 2 of the License, or *
15		* (at your option) any later version. *
16		* *
17		* This program is distributed in the hope that it will be useful, *
18		* but WITHOUT ANY WARRANTY; without even the implied warranty of *
19		* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
20		* GNU General Public License for more details. *
21		* *
22		* You should have received a copy of the GNU General Public License *
23		* along with this program; if not, write to the Free Software *
24		* Foundation, Inc., 51 Franklin Street, Fifth Floor, *
25		* Boston, MA 02110-1301 USA *
26		* *
27		***************************************************************************/
28
29		#ifndef CORRELATIONCOEFFICIENT_H
30		#define CORRELATIONCOEFFICIENT_H
31
32		#include "backend/core/AbstractPart.h"
33		#include "GeneralTest.h"
34		#include "backend/lib/macros.h"
35
36		class CorrelationCoefficientView;
	sgerlachUnsubmitted Done are all of them needed? sgerlach: are all of them needed?
	devanshuagarwalAuthorUnsubmitted Done Thanks, I removed unnecessary ones. devanshuagarwal: Thanks, I removed unnecessary ones.
37		class Spreadsheet;
38		class QString;
39		class Column;
40		class QVBoxLayout;
41		class QLabel;
42
43		class CorrelationCoefficient : public GeneralTest {
44		Q_OBJECT
45
46		public:
47		explicit CorrelationCoefficient(const QString& name);
48		~CorrelationCoefficient() override;
49
50		enum Test{
51		Pearson,
52		Kendall,
53		Spearman
54		};
55		double correlationValue();
56		QWidget* view() const override;
57
58		void performTest(Test m_test, bool categoricalVariable = true);
59		private:
60		void performPearson(bool categoricalVariable);
61		void performKendall();
62		void performSpearman();
63
64		int findDiscordants(int* ranks, int start, int end);
65
66		void convertToRanks(const Column* col, int N, QMap<double, int> &ranks);
67		void convertToRanks(const Column* col, QMap<double, int> &ranks);
68
69		double m_correlationValue;
70		};
71
72		#endif // CORRELATIONCOEFFICIENT_H