joel/largeRCRF-Java
1
0
Fork 0
Code Issues 3 Pull requests Releases Wiki Activity

Add functionality to restart tree training where previously left off.

Browse source
This commit is contained in:
Joel Therrien 2018-08-09 16:34:10 -07:00
parent 6d65d48844
commit e92abdab13
1 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/main/java/ca/joeltherrien/randomforest/tree/ForestTrainer.java
View file

@ -8,6 +8,7 @@ import lombok.AccessLevel;
import lombok.AllArgsConstructor;
import lombok.Builder;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
@ -122,10 +123,19 @@ public class ForestTrainer<Y, TO, FO> {
public void trainParallelOnDisk(int threads){
final ExecutorService executorService = Executors.newFixedThreadPool(threads);
final AtomicInteger treeCount = new AtomicInteger(0); // tracks how many trees are finished
// First we need to see how many trees there currently are
final File folder = new File(saveTreeLocation);
if(!folder.isDirectory()){
throw new IllegalArgumentException("Tree directory must be a directory!");
}
for(int j=0; j<ntree; j++){
final File[] treeFiles = folder.listFiles(((file, s) -> s.endsWith(".tree")));
final ExecutorService executorService = Executors.newFixedThreadPool(threads);
final AtomicInteger treeCount = new AtomicInteger(treeFiles.length); // tracks how many trees are finished
for(int j=treeCount.get(); j<ntree; j++){
final Runnable worker = new TreeSavedWorker(data, "tree-" + (j+1) + ".tree", treeCount);
executorService.execute(worker);
}