Added member variables to LoggingEventArgs to control logging

src/Microsoft.ML.Data/LoggingEventArgs.cs

@@ -3,6 +3,7 @@
 // See the LICENSE file in the project root for more information.
 
 using System;
+using Microsoft.ML.Runtime;
 
 namespace Microsoft.ML
 {
@@ -20,6 +21,29 @@ public LoggingEventArgs(string message)
             Message = message;
         }
 
+        /// <summary>
+        /// Initializes a new instane of <see cref="LoggingEventArgs"/> class that includes the kind and source of the message
+        /// </summary>
+        /// <param name="message"> The message being logged </param>
+        /// <param name="kind"> The type of message <see cref="ChannelMessageKind"/> </param>
+        /// <param name="source"> The source of the message </param>
+        public LoggingEventArgs(string message, ChannelMessageKind kind, string source)
+        {
+            Message = message;
+            Kind = kind;
+            Source = source;
+        }
+
+        /// <summary>
+        /// Gets the source component of the event
+        /// </summary>
+        public string Source { get; }
+
+        /// <summary>
+        /// Gets the type of message
+        /// </summary>
+        public ChannelMessageKind Kind { get; }
+
         /// <summary>
         /// Gets the message being logged.
         /// </summary>

src/Microsoft.ML.Data/MLContext.cs

@@ -131,9 +131,7 @@ private void ProcessMessage(IMessageSource source, ChannelMessage message)
             if (log == null)
                 return;
 
-            var msg = $"[Source={source.FullName}, Kind={message.Kind}] {message.Message}";
-
-            log(this, new LoggingEventArgs(msg));
+            log(this, new LoggingEventArgs(message.Message, message.Kind, source.FullName));
         }
 
         string IExceptionContext.ContextDescription => _env.ContextDescription;

test/Microsoft.ML.TestFramework/BaseTestBaseline.cs

@@ -109,7 +109,8 @@ protected override void Initialize()
 
         private void LogTestOutput(object sender, LoggingEventArgs e)
         {
-            Output.WriteLine(e.Message);
+            var msg = $"[Source={e.Source}, Kind={e.Kind}] {e.Message}";
+            Output.WriteLine(msg);
         }
 
         // This method is used by subclass to dispose of disposable objects