Exemplo n.º 1
0
        private static void Test1_5d()
        {
            double[] data = ReadData(@"data\input1.txt");
            EM       em   = new EM(data, 3);

            em.Initialize(2);
            em.Run();
            PrintEM(em);
        }
Exemplo n.º 2
0
        private static string PrintEM(EM em, bool printToConsole = true)
        {
            int           padSize      = 15;
            string        numberFormat = "F6";
            StringBuilder sb           = new StringBuilder();

            // Print Mean, LL, BIC

            for (int j = 0; j < em.K; j++)
            {
                sb.Append(("mu" + (j + 1)).PadLeft(padSize));
            }
            sb.Append("LogLik".PadLeft(padSize));
            sb.Append("BIC".PadLeft(padSize));

            sb.AppendLine();

            for (int i = 0; i < em.Means.Count; i++)
            {
                for (int j = 0; j < em.K; j++)
                {
                    sb.Append(em.Means[i][j].ToString(numberFormat).PadLeft(padSize));
                }
                sb.Append(em.LogLikelihoods[i].ToString(numberFormat).PadLeft(padSize));
                sb.Append(em.BICs[i].ToString(numberFormat).PadLeft(padSize));
                sb.AppendLine();
            }

            // Print xi
            sb.AppendLine();
            sb.Append("xi".PadLeft(padSize));
            for (int j = 0; j < em.K; j++)
            {
                sb.Append(("P(cls " + (j + 1) + " | xi)").PadLeft(padSize));
            }
            sb.AppendLine();

            for (int i = 0; i < em.X.Length; i++)
            {
                sb.Append(("[" + (i + 1) + "]").PadLeft(6));
                sb.Append(em.X[i].ToString().PadLeft(padSize - 6));
                for (int j = 0; j < em.K; j++)
                {
                    sb.Append(em.E_Steps.Last()[i, j].ToString("e6").PadLeft(padSize));
                }
                sb.AppendLine();
            }

            if (printToConsole)
            {
                Console.WriteLine(sb.ToString());
            }

            return(sb.ToString());
        }
Exemplo n.º 3
0
        private static void Test1_5a()
        {
            double[] data = ReadData(@"data\input1.txt");
            EM       em   = new EM(data, 5);

            em.Initialize(new double[5] {
                35, 12, 46, 22, 45
            });
            em.Run();
            PrintEM(em);
        }
Exemplo n.º 4
0
        private static void Test1_3()
        {
            double[] data = ReadData(@"data\input1.txt");
            EM       em   = new EM(data, 3);

            em.Initialize(new double[3] {
                21, 46, 55
            });
            em.Run();
            PrintEM(em);
        }
Exemplo n.º 5
0
        private static void Test1_5b()
        {
            double[] data = ReadData(@"data\input1.txt");
            EM       em   = new EM(data, 5);

            em.Initialize(new double[5] {
                9, 10, 46, 49, 57
            });
            em.Run();
            PrintEM(em);
        }
Exemplo n.º 6
0
        private static void BestBIC(string inputFile, int randomSeed)
        {
            double[] data = ReadData(inputFile);

            List <EM> results = new List <EM>();

            int maxK            = 5;
            int highestBICIndex = 0;

            for (int k = 0; k < maxK; k++)
            {
                EM em = new EM(data, k + 1);
                em.Initialize(randomSeed);
                em.Run();
                results.Add(em);

                // Keep track of max BIC
                if (em.BICs.Last() > results[highestBICIndex].BICs.Last())
                {
                    highestBICIndex = k;
                }
            }

            StringBuilder sb = new StringBuilder();

            sb.Append("k".PadLeft(6));
            sb.Append("BIC".PadLeft(14));
            sb.AppendLine();
            for (int i = 0; i < maxK; i++)
            {
                sb.Append((i + 1).ToString().PadLeft(6));
                sb.Append(results[i].BICs.Last().ToString("F6").PadLeft(14));

                if (i == highestBICIndex)
                {
                    sb.Append("  <--- Highest BIC");
                }

                sb.AppendLine();
            }
            sb.AppendLine();

            sb.Append(PrintEM(results[highestBICIndex], false));

            Console.WriteLine(sb.ToString());

            File.WriteAllText("results.txt", sb.ToString());
        }