You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
* fix up docs for MLContext
* some more fixes
* text class and sentence similarity trainers
---------
Co-authored-by: Genevieve Warren <[email protected]>
Copy file name to clipboardExpand all lines: src/Microsoft.ML.TorchSharp/NasBert/SentenceSimilarityTrainer.cs
+11-10
Original file line number
Diff line number
Diff line change
@@ -27,31 +27,32 @@
27
27
namespaceMicrosoft.ML.TorchSharp.NasBert
28
28
{
29
29
/// <summary>
30
-
/// The <see cref="IEstimator{TTransformer}"/> for training a Deep Neural Network(DNN) to classify text.
30
+
/// Represents the <see cref="IEstimator{TTransformer}"/> for training a Deep Neural Network(DNN) to determine sentence similarity.
31
31
/// </summary>
32
32
/// <remarks>
33
33
/// <format type="text/markdown"><).
35
35
///
36
-
/// ### Input and Output Columns
37
-
/// The input label column data must be type<xref:System.Single> type and the sentence columns must be of type<xref:Microsoft.ML.Data.TextDataViewType>.
36
+
/// ### Input and output columns
37
+
/// The input label column data must be type<xref:System.Single> and the sentence columns must be of type<xref:Microsoft.ML.Data.TextDataViewType>.
38
38
///
39
39
/// This trainer outputs the following columns:
40
40
///
41
-
/// | Output Column Name | Column Type | Description|
41
+
/// | Output column name | Column type | Description|
42
42
/// | -- | -- | -- |
43
-
/// | `Score` | <xref:System.Single> | The degree of similarity between the 2 sentences. |
44
-
/// ### Trainer Characteristics
45
-
/// | | |
43
+
/// | `Score` | <xref:System.Single> | The degree of similarity between the two sentences. |
44
+
///
45
+
/// ### Trainer characteristics
46
+
/// | Characteristic | Value |
46
47
/// | -- | -- |
47
-
/// | Machine learning task | Rregression |
48
+
/// | Machine learning task | Regression |
48
49
/// | Is normalization required? | No |
49
50
/// | Is caching required? | No |
50
51
/// | Required NuGet in addition to Microsoft.ML | Microsoft.ML.TorchSharp and libtorch-cpu or libtorch-cuda-11.3 or any of the OS specific variants. |
51
52
/// | Exportable to ONNX | No |
52
53
///
53
-
/// ### Training Algorithm Details
54
-
/// Trains a Deep Neural Network(DNN) by leveraging an existing pre-trained NAS-BERT roBERTa model for the purpose of determining sentence similarity.
54
+
/// ### Training algorithm details
55
+
/// Trains a Deep Neural Network(DNN) by leveraging an existing, pretrained NAS-BERT roBERTa model for the purpose of determining sentence similarity.
Copy file name to clipboardExpand all lines: src/Microsoft.ML.TorchSharp/NasBert/TextClassificationTrainer.cs
+11-10
Original file line number
Diff line number
Diff line change
@@ -28,32 +28,33 @@
28
28
namespaceMicrosoft.ML.TorchSharp.NasBert
29
29
{
30
30
/// <summary>
31
-
/// The <see cref="IEstimator{TTransformer}"/> for training a Deep Neural Network(DNN) to classify text.
31
+
/// The <see cref="IEstimator{TTransformer}"/> for training a Deep Neural Network(DNN) to classify text.
32
32
/// </summary>
33
33
/// <remarks>
34
34
/// <format type="text/markdown"><).
36
36
///
37
-
/// ### Input and Output Columns
38
-
/// The input label column data must be [key](xref:Microsoft.ML.Data.KeyDataViewType) type and the sentence columns must be of type<xref:Microsoft.ML.Data.TextDataViewType>.
37
+
/// ### Input and output columns
38
+
/// The input label column data must be [key](xref:Microsoft.ML.Data.KeyDataViewType) type and the sentence columns must be of type<xref:Microsoft.ML.Data.TextDataViewType>.
39
39
///
40
40
/// This trainer outputs the following columns:
41
41
///
42
-
/// | Output Column Name | Column Type | Description|
42
+
/// | Output column name | Column type | Description|
43
43
/// | -- | -- | -- |
44
-
/// | `PredictedLabel` | [key](xref:Microsoft.ML.Data.KeyDataViewType) type | The predicted label's index. If its value is i, the actual label would be the i-th category in the key-valued input label type. |
45
-
/// | `Score` | Vector of<xref:System.Single> | The scores of all classes.Higher value means higher probability to fall into the associated class. If the i-th element has the largest value, the predicted label index would be i.Note that i is zero-based index. |
46
-
/// ### Trainer Characteristics
47
-
/// | | |
44
+
/// | `PredictedLabel` | [key](xref:Microsoft.ML.Data.KeyDataViewType) type | The predicted label's index. If its value is `i`, the actual label would be the `i`-th category in the key-valued input label type. |
45
+
/// | `Score` | Vector of<xref:System.Single> | The scores of all classes. Higher value means higher probability to fall into the associated class. If the `i`-th element has the largest value, the predicted label index would be `i`. Note that `i` is a zero-based index. |
/// | Required NuGet in addition to Microsoft.ML | Microsoft.ML.TorchSharp and libtorch-cpu or libtorch-cuda-11.3 or any of the OS specific variants. |
53
54
/// | Exportable to ONNX | No |
54
55
///
55
-
/// ### Training Algorithm Details
56
-
/// Trains a Deep Neural Network(DNN) by leveraging an existing pre-trained NAS-BERT roBERTa model for the purpose of classifying text.
56
+
/// ### Training algorithm details
57
+
/// Trains a Deep Neural Network(DNN) by leveraging an existing, pretrained NAS-BERT roBERTa model for the purpose of classifying text.
0 commit comments