示例#1
0
        public bool TrainModel(string dbFileName)
        {
            var dataFileName  = $"{dbFileName}.csv";
            var modelFileName = "jcids.mdl";

            var networkData = new NetworkDeviceManager().ToCSV();

            File.WriteAllText(dataFileName, networkData);

            var textReader = MlContext.Data.LoadFromTextFile(dataFileName);

            var pipeline = MlContext.Transforms.Text.FeaturizeText("Content", "Features")
                           .Append(MlContext.BinaryClassification.Trainers.FastTree(numLeaves: 2, numTrees: 10, minDatapointsInLeaves: 1));

            var trainedModel = pipeline.Fit(textReader);

            using (var fs = File.Create(modelFileName))
            {
                trainedModel.SaveTo(MlContext, fs);
            }

            return(true);
        }
示例#2
0
 public DevicesController(NetworkDeviceManager deviceManager)
 {
     _deviceManager = deviceManager;
 }