hama-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From edwardy...@apache.org
Subject svn commit: r1526167 - /hama/trunk/examples/src/test/java/org/apache/hama/examples/NeuralNetworkTest.java
Date Wed, 25 Sep 2013 12:40:53 GMT
Author: edwardyoon
Date: Wed Sep 25 12:40:52 2013
New Revision: 1526167

URL: http://svn.apache.org/r1526167
Log:
HAMA-804: Fix rat check fail

Modified:
    hama/trunk/examples/src/test/java/org/apache/hama/examples/NeuralNetworkTest.java

Modified: hama/trunk/examples/src/test/java/org/apache/hama/examples/NeuralNetworkTest.java
URL: http://svn.apache.org/viewvc/hama/trunk/examples/src/test/java/org/apache/hama/examples/NeuralNetworkTest.java?rev=1526167&r1=1526166&r2=1526167&view=diff
==============================================================================
--- hama/trunk/examples/src/test/java/org/apache/hama/examples/NeuralNetworkTest.java (original)
+++ hama/trunk/examples/src/test/java/org/apache/hama/examples/NeuralNetworkTest.java Wed
Sep 25 12:40:52 2013
@@ -43,10 +43,12 @@ public class NeuralNetworkTest {
     this.testNeuralNetworkTraining();
 
     String dataPath = "src/test/resources/neuralnets_classification_test.txt";
-    String modelPath = "tmp/neuralnets.model";
-    String resultPath = "tmp/neuralnets.txt";
+    String modelPath = "/tmp/neuralnets.model";
+    String resultPath = "/tmp/neuralnets.txt";
     String mode = "label";
+    Configuration conf = new Configuration();
     try {
+      FileSystem fs = FileSystem.get(conf);
       NeuralNetwork
           .main(new String[] { mode, modelPath, dataPath, resultPath });
 
@@ -76,6 +78,9 @@ public class NeuralNetworkTest {
         }
       }
       System.out.printf("Precision: %f\n", correct / total);
+      fs.delete(new Path(resultPath), true);
+      fs.delete(new Path(modelPath), true);
+
     } catch (Exception e) {
       e.printStackTrace();
     }
@@ -84,14 +89,15 @@ public class NeuralNetworkTest {
   private void testNeuralNetworkTraining() {
     String mode = "train";
     String strTrainingDataPath = "src/test/resources/neuralnets_classification_training.txt";
-    String strSequenceTrainingDataPath = "tmp/test-neuralnets.data";
+    String strSequenceTrainingDataPath = "/tmp/test-neuralnets.data";
     int featureDimension = 8;
     int labelDimension = 1;
 
     Path sequenceTrainingDataPath = new Path(strSequenceTrainingDataPath);
     Configuration conf = new Configuration();
+    FileSystem fs;
     try {
-      FileSystem fs = FileSystem.get(conf);
+      fs = FileSystem.get(conf);
       SequenceFile.Writer writer = new SequenceFile.Writer(fs, conf,
           sequenceTrainingDataPath, LongWritable.class, VectorWritable.class);
       BufferedReader br = new BufferedReader(
@@ -113,10 +119,13 @@ public class NeuralNetworkTest {
       e1.printStackTrace();
     }
 
-    String modelPath = "tmp/neuralnets.model";
+    String modelPath = "/tmp/neuralnets.model";
     try {
       NeuralNetwork.main(new String[] { mode, strSequenceTrainingDataPath,
           modelPath, "" + featureDimension, "" + labelDimension });
+      fs = FileSystem.get(conf);
+      fs.delete(new Path(strSequenceTrainingDataPath), true);
+      fs.delete(new Path(modelPath), true);
     } catch (Exception e) {
       e.printStackTrace();
     }



Mime
View raw message