Esempio n. 1
0
        public static Word2VecTestClass GetFruitAndJuiceCbowPreset()
        {
            var testInput =
                "drink^juice|apple,eat^apple|orange,drink^juice|rice,drink^milk|juice,drink^rice|milk,drink^milk|water,orange^apple|juice,apple^drink|juice,rice^drink|milk,milk^water|drink,water^juice|drink,juice^water|drink";
            var testSubject = new Word2VecTestClass
            {
                Size         = 5,
                Sample       = 0,
                LearningRate = 0.2D,
                IsCbow       = true,
                IsDebugMode  = true
            };

            testSubject.InitTest(testInput);

            return(InitTestClassVectors(testSubject));
        }
Esempio n. 2
0
        public void TestInitTest()
        {
            var testInput   = "drink^juice|apple,eat^apple|orange,drink^juice|rice,drink^milk|juice,drink^rice|milk,drink^milk|water,orange^apple|juice,apple^drink|juice,rice^drink|milk,milk^water|drink,water^juice|drink,juice^water|drink";
            var testSubject = new Word2VecTestClass();

            testSubject.InitTest(testInput);
            //"apple", "drink", "juice"
            var testResult = testSubject.ReadNextWord();

            Assert.IsNotNull(testResult);
            Assert.IsNotNull(testResult.TargetWord);
            Assert.IsNotNull(testResult.ContextWords);
            Assert.AreEqual(2, testResult.ContextWords.Count);
            Assert.AreEqual("apple", testResult.TargetWord.Word);
            Assert.IsTrue(testResult.ContextWords.Any(v => v.Word == "drink"));
            Assert.IsTrue(testResult.ContextWords.Any(v => v.Word == "juice"));
        }