Пример #1
0
        public void PostMultiple_positiveText_True()
        {
            //Arrange
            MLContext      mlContext = new MLContext();
            DataViewSchema predictionPipelineSchema;
            PredictionEngine <SentimentData, SentimentPrediction> predictionEngine = mlContext
                                                                                     .Model.CreatePredictionEngine <SentimentData, SentimentPrediction>(mlContext
                                                                                                                                                        .Model.Load("C:/Users/iRebeca/facultate/.NET/.NetProject/TSense/Tests/MlModels/sentiment_model.zip", out predictionPipelineSchema));
            SentimentController sentiment     = new SentimentController(predictionEngine);
            SentimentData       sentimentData = new SentimentData();

            sentimentData.SentimentText = "It was great";
            List <SentimentData> sentiments = new List <SentimentData>()
            {
                sentimentData
            };

            //Act
            var result     = (OkObjectResult)sentiment.PostMultiple(sentiments).Result;
            var prediction = (ML_Server.DataModels.Sentiment)result.Value;

            //Assert
            Assert.IsInstanceOfType(result.Value, typeof(ML_Server.DataModels.Sentiment));
            Assert.AreEqual(true, prediction.Prediction, prediction.Probability.ToString());
        }
Пример #2
0
        public void Post_negativeText_False()
        {
            //Arrange
            MLContext      mlContext = new MLContext();
            DataViewSchema predictionPipelineSchema;
            ITransformer   predictionPipeline = mlContext
                                                .Model.Load("C:/Users/iRebeca/facultate/.NET/.NetProject/TSense/Tests/MlModels/sentiment_model.zip", out predictionPipelineSchema);
            PredictionEngine <SentimentData, SentimentPrediction> predictionEngine = mlContext
                                                                                     .Model.CreatePredictionEngine <SentimentData, SentimentPrediction>(predictionPipeline);
            SentimentController sentiment     = new SentimentController(predictionEngine);
            SentimentData       sentimentData = new SentimentData();

            sentimentData.SentimentText = "It was bad";

            //Act
            var result     = (OkObjectResult)sentiment.Post(sentimentData).Result;
            var prediction = (SentimentPrediction)result.Value;

            //Assert
            Assert.IsInstanceOfType(result.Value, typeof(SentimentPrediction));
            Assert.AreEqual(prediction.Prediction, false);
        }