Add pattern match

nicholasmerci · Aug 22, 2020 · a311ea2 · a311ea2
1 parent 24c729d
commit a311ea2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/MLNetExamples/SentimentAnalysis/Program.cs b/MLNetExamples/SentimentAnalysis/Program.cs
@@ -21,13 +21,28 @@ static void Main(string[] args)
 
             var prediction = predictionEngine.Predict(new SentimentData { Text = "I would buy MSFT shares." });
 
-            Console.WriteLine($"Prediction - {prediction.Prediction} with score - {prediction.Score}");
+            Console.WriteLine($"Prediction - {prediction.Prediction} with probability - {prediction.Probability}");
 
             var newPrediction = predictionEngine.Predict(new SentimentData { Text = "TWTR may close at a low today." });
 
-            Console.WriteLine($"Prediction - {newPrediction.Prediction} with score - {newPrediction.Score}");
-
-            Console.ReadLine();
+            Console.WriteLine($"Prediction - {newPrediction.Prediction} with probability - {newPrediction.Probability}");
+
+            var anotherPrediction = predictionEngine.Predict(new SentimentData { Text = "TSLA is at an all time high." });
+
+            switch (anotherPrediction.Probability)
+            {
+                case float p when p < .5:
+                    Console.WriteLine($"TSLA sentiment is negative with probability {p}");
+                    break;
+                case float p when p >= .5 && p <= .7:
+                    Console.WriteLine($"TSLA sentiment is neutral with probability {p}");
+                    break;
+                case float p when p > .7:
+                    Console.WriteLine($"TSLA sentiment is positive with probability {p}");
+                    break;
+                default:
+                    break;
+            }
         }
     }
 }
diff --git a/MLNetExamples/SentimentAnalysis/SentimentAnalysis.csproj b/MLNetExamples/SentimentAnalysis/SentimentAnalysis.csproj
@@ -2,11 +2,11 @@
 
   <PropertyGroup>
     <OutputType>Exe</OutputType>
-    <TargetFramework>netcoreapp2.2</TargetFramework>
+    <TargetFramework>netcoreapp3.1</TargetFramework>
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Microsoft.ML" Version="1.5.2-29213-2" />
+    <PackageReference Include="Microsoft.ML" Version="1.5.2-29219-1" />
   </ItemGroup>
 
   <ItemGroup>