Пример #1
0
        public void LimitedMemoryBgfsOverrides()
        {
            var testSubject = TrainLimitedMemoryBfgsOpt.Train(TrainType.MaximumEntropyModel)
                              .SetClipGradient()
                              .SetHistorySize(101)
                              .MaxLinearSearchIterations(202)
                              .Get();

            testSubject.Clip.Should().BeTrue();
            testSubject.HistSz.Should().Be(101);
            testSubject.MaxLs.Should().Be(202);
        }
Пример #2
0
        public void GrobidDateModelTest()
        {
            var opt = TrainLimitedMemoryBfgsOpt.Train()
                      .SetStopEpsilon(1e-5)
                      .SetStopWindowSize(20)
                      .SetMaxIterations(2000)
                      .Get();

            var trainer = new WapitiTrainer();
            var result  = trainer.Train(
                opt,
                "date.training.txt".GetResourceStream(),
                "date.template.txt".GetResourceStream());

            Approvals.Verify(result);
        }
Пример #3
0
        public void CommonOverrides()
        {
            var testSubject = TrainLimitedMemoryBfgsOpt.Train()
                              .SetL1Penalty(1.01)
                              .SetL2Penalty(2.02)
                              .SetCompact()
                              .SetSparse()
                              .SetConvergenceWindowSize(303)
                              .SetStopWindowSize(404)
                              .SetStopEpsilon(5.05)
                              .Get();

            testSubject.Rho1.Should().Be(1.01);
            testSubject.Rho2.Should().Be(2.02);
            testSubject.Compact.Should().BeTrue();
            testSubject.Sparse.Should().BeTrue();
            testSubject.ObjWin.Should().Be(303);
            testSubject.StopWin.Should().Be(404);
            testSubject.StopEps.Should().Be(5.05);
        }
Пример #4
0
        public void LimitedMemoryBgfsDefaults()
        {
            var testSubject = TrainLimitedMemoryBfgsOpt.Train().Get();

            testSubject.AssertDefaultTrains(TrainType.ConditionalRandomFields, TrainAlgo.LimitedMemoryBfgs);
        }
Пример #5
0
        public void LimitedMemoryBgfsMemm()
        {
            var testSubject = TrainLimitedMemoryBfgsOpt.Train(TrainType.MaximumEntropyMarkovModel).Get();

            testSubject.AssertDefaultTrains(TrainType.MaximumEntropyMarkovModel, TrainAlgo.LimitedMemoryBfgs);
        }