class definition

source/sample/transformer/T2TSearch.h

@@ -26,6 +26,22 @@
 
 namespace transformer
 {
+
+/* The class orgnizes the search process. It calls predictors to generate 
+   distributions of the predictions and prunes the search space by beam pruning.
+   It results in a graph where each path respresents a translation hypothsis.
+   The output can be the path with the highest model score. */
+class T2TSearch
+{
+public:
+    /* constructor */
+    T2TSearch() {};
+
+    /* de-constructor */
+    ~T2TSearch() {};
+
+};
+
 }
 
 #endif
\ No newline at end of file

source/sample/transformer/T2TTester.h

@@ -27,6 +27,13 @@
 
 namespace transformer
 {
+
+/* This class translates test sentences with a trained model. */
+class T2TTester
+{
+public:
+};
+
 }
 
 #endif
\ No newline at end of file