Пример #1
0
        public void GAMClassificationEstimator()
        {
            var(pipe, dataView) = GetBinaryClassificationPipeline();

            var trainer = new BinaryClassificationGamTrainer(Env, new BinaryClassificationGamTrainer.Options
            {
                GainConfidenceLevel = 0,
                NumIterations       = 15,
            });
            var pipeWithTrainer = pipe.Append(trainer);

            TestEstimatorCore(pipeWithTrainer, dataView);

            var transformedDataView = pipe.Fit(dataView).Transform(dataView);
            var model = trainer.Train(transformedDataView, transformedDataView);

            Done();
        }
Пример #2
0
        public void GAMClassificationEstimator()
        {
            var(pipe, dataView) = GetBinaryClassificationPipeline();

            var trainer = new BinaryClassificationGamTrainer(Env, "Label", "Features", advancedSettings: s =>
            {
                s.GainConfidenceLevel = 0;
                s.NumIterations       = 15;
            });
            var pipeWithTrainer = pipe.Append(trainer);

            TestEstimatorCore(pipeWithTrainer, dataView);

            var transformedDataView = pipe.Fit(dataView).Transform(dataView);
            var model = trainer.Train(transformedDataView, transformedDataView);

            Done();
        }