void main() {
	cout << "1. Information" << endl << "2. Entropy" << endl << endl << "Make your choice." << endl;
	int choice(0);
	float input2(-1.0f);
	int input(-1);
	Information information = Information();
	Entropy entropy = Entropy();
	cin >> choice;
	switch (choice)
	{
	case 1:
		while (input !=0)
		{
			cout << "Type an integer count for the classe. Type 0 to end.";
			cin >> input;
			if (input > 0)
			{
				information.AddClass(input);
			}
		}
		cout << "Result :" << information.ComputeInformation() << endl;
		break;
	case 2:

		while (input2 != 0.0f)
		{
			cout << "Type a float probability for the classe. Type 0 to end.";
			cin >> input2;
			if (input2 > 0.0f)
			{
				entropy.AddProbability(input2);
			}
		}
		cout << "Result :" << entropy.ComputeEntropy() <<endl;
		break;
	}
	system("pause");
}