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)); }
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")); }