Exemplo n.º 1
0
        Intent PredictSingleItem(ClassificationData statement, bool debug = false)
        {
            PredictionEngine <ClassificationData, ClassificationPrediction> predictionFunction =
                _mlContext.Model.CreatePredictionEngine <ClassificationData, ClassificationPrediction>(_model);


            var resultprediction = predictionFunction.Predict(statement);

            Intent intent = new Intent();

            intent.Name  = _datasetName;
            intent.Score = resultprediction.Probability;

            if (debug)
            {
                Console.WriteLine("-------- Prediction of " + _datasetName + " model --------");
                Console.WriteLine("Query: " + resultprediction.Content
                                  + " | Prediction (" + _datasetName + "): " + Convert.ToBoolean(resultprediction.Prediction)
                                  + " | Probability: " + resultprediction.Probability);

                Console.WriteLine("----------------------------------------------------");
                Console.WriteLine();
            }

            return(intent);
        }
Exemplo n.º 2
0
        public Intent Classify(string query, bool debug = false)
        {
            ClassificationData statement = new ClassificationData {
                Content = query
            };

            return(PredictSingleItem(statement, debug));
        }