updates

source/train/TTrain.cpp

@@ -122,7 +122,7 @@ void TTDataLoader::SetBatchSize(int myBatchSize)
 /* start the process */
 bool TTDataLoader::Start()
 {
-    FILE * file = fopen(fileName, "wb");
+    file = fopen(fileName, "wb");
     CheckNTErrors(file, "Cannot open the file");
 
     /* skip the first line */
@@ -144,6 +144,8 @@ bool TTDataLoader::End()
 /* get a batch of samples */
 bool TTDataLoader::GetBatch(XList * args)
 {
+    CheckNTErrors(file != NULL, "No input file specificed!");
+
     XTensor * input = NULL;
     XTensor * gold = NULL;
     XTensor * output = NULL;

source/train/XWorkerJob.h

@@ -48,6 +48,9 @@ protected:
 
     /* the output tensors of the model */
     XList outputs;
+
+    /* the gold standard  */
+    XList gold;
     
 public:
 
@@ -71,6 +74,9 @@ public:
 
     /* get the output list */
     XList * GetOutput();
+    
+    /* get the gold standard */
+    XList * GetGold();
 
     /* add a new job of model refreshment */
     bool AddJobRefresh(XModel * myModel);