Java Deep Learning Projects

Wrapping everything up by executing the main() method

Let's wrap the overall discussion by watching the performance of our model programmatically (see the main YelpImageClassifier.java class):

public class YelpImageClassifier {    public static void main(String[] args) throws IOException {        Map<String, Set<Integer>> labMap = readBusinessLabels("Yelp/labels/train.csv");                Map<Integer, String> businessMap = readBusinessToImageLabels("Yelp/labels                                                                      /train_photo_to_biz_ids.csv");        List<String> businessIds = businessMap.entrySet().stream().map(e ->                                                     e.getValue()).distinct().collect(Collectors.toList());        // 100 images        List<String> imgs = getImageIds("Yelp/images/train/", businessMap, businessIds).subList(0, 100);  System.out.println("Image ID retreival ...