Sentence Similarity (#6390)

Author: michaelgsharp

src/Microsoft.ML.TorchSharp/NasBert/Models/BaseModel.cs

@@ -13,7 +13,7 @@ namespace Microsoft.ML.TorchSharp.NasBert.Models
 {
     internal abstract class BaseModel : BaseModule
     {
-        protected readonly TextClassificationTrainer.Options Options;
+        protected readonly NasBertTrainer.Options Options;
         public BertTaskType HeadType => Options.TaskType;
 
         protected readonly TransformerEncoder Encoder;
@@ -24,7 +24,7 @@ internal abstract class BaseModel : BaseModule
         public abstract BaseHead GetHead();
 #pragma warning restore CA1024 // Use properties where appropriate
 
-        protected BaseModel(TextClassificationTrainer.Options options, int padIndex, int symbolsCount)
+        protected BaseModel(NasBertTrainer.Options options, int padIndex, int symbolsCount)
             : base(nameof(BaseModel))
         {
             Options = options ?? throw new ArgumentNullException(nameof(options));

src/Microsoft.ML.TorchSharp/NasBert/Models/TextClassificationModel.cs renamed to src/Microsoft.ML.TorchSharp/NasBert/Models/NasBertModel.cs

@@ -9,13 +9,13 @@
 
 namespace Microsoft.ML.TorchSharp.NasBert.Models
 {
-    internal sealed class TextClassificationModel : BaseModel
+    internal sealed class NasBertModel : BaseModel
     {
         private readonly PredictionHead _predictionHead;
 
         public override BaseHead GetHead() => _predictionHead;
 
-        public TextClassificationModel(TextClassificationTrainer.Options options, int padIndex, int symbolsCount, int numClasses)
+        public NasBertModel(NasBertTrainer.Options options, int padIndex, int symbolsCount, int numClasses)
             : base(options, padIndex, symbolsCount)
         {
             _predictionHead = new PredictionHead(