FIX:win libsvm convert tool & predict batch file

kanyun-inc · May 20, 2017 · 3ce9925 · 3ce9925
1 parent 9283552
commit 3ce9925
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/bin/win_libsvm_convert_2_ytklearn.bat b/bin/win_libsvm_convert_2_ytklearn.bat
@@ -1,6 +1,6 @@
 echo on
 @Rem binary_classification@label1,label2, multi_classification@label1,label2,..., regression
-set mode= ???
+set mode=???
 
 set x_delim="###"
 set y_delim=","
@@ -11,6 +11,7 @@ set fs_scheme="???"
 set libsvm_data_path="???"
 set ytklearn_data_path="???"
 
+cd ..
 @start /b cmd /c java -Xmx1000m -XX:-OmitStackTraceInFastThrow -cp lib/* -Dlog4j.configuration=file:config/log4j.properties com.fenbi.ytklearn.utils.LibsvmConvertTool %mode% %x_delim% %y_delim% %features_delim% %feature_name_val_delim% %fs_scheme% %libsvm_data_path% %ytklearn_data_path%
 
 pause
diff --git a/bin/win_predict.bat b/bin/win_predict.bat
@@ -23,6 +23,7 @@ set eval_metric="auc,mae"
 @Rem value or leafid
 set predict_type=value
 
+cd ..
 @start /b cmd /c java -Xmx1000m -XX:-OmitStackTraceInFastThrow -cp lib/* -Dlog4j.configuration=file:config/log4j.properties com.fenbi.ytklearn.predictor.Predicts %config_path% %model_name% %file_name% %transform% %transform_script_path% %resultSaveMode% %resultFileSuffix% %max_error_tol% %eval_metric% %predict_type%
 
 pause
diff --git a/src/main/java/com/fenbi/ytklearn/fs/LocalFileSystem.java b/src/main/java/com/fenbi/ytklearn/fs/LocalFileSystem.java
@@ -49,7 +49,10 @@ public Reader getReader(String path) throws IOException {
 
     @Override
     public Writer getWriter(String path) throws IOException {
-        mkdirs(new File(path).getParentFile().getAbsolutePath());
+        File parentFile = new File(path).getParentFile();
+        if (parentFile != null) {
+            mkdirs(parentFile.getAbsolutePath());
+        }
         return new PrintWriter(path);
     }