private void updateInstanceAndPutInField( Instance instance, int correctClass, int predictedClassIndex) { Segment segment = toSegment(instance); String classAsString = instance.classAttribute().value(predictedClassIndex); int predictedLabelId = Integer.parseInt(classAsString); segment.setPredictedLabelDetail(labelMapReader.getSchema().get(predictedLabelId)); putInField(segment, correctClass, predictedClassIndex); }
private String getHeaders(List<Segment> segments) { if (segments.size() == 0) { return ""; } Segment segment = segments.get(0); StringBuilder output = new StringBuilder(); output.append("device_info_serial,"); output.append("date_time,"); output.append("lon,"); output.append("lat,"); output.append("alt,"); output.append("class_id,"); for (String feature : segment.getFeatureNames()) { output.append(feature + ","); } output.append("set\n"); return output.toString(); }
private String getCsv(List<Segment> segments, String set) { StringBuilder output = new StringBuilder(); for (Segment segment : segments) { output.append(segment.getDeviceId() + ","); output.append(segment.getTimeStamp().toString(Constants.DATE_TIME_PATTERN_ISO8601) + ","); output.append(segment.getLongitude() + ","); output.append(segment.getLatitude() + ","); output.append(segment.getAltitude() + ","); output.append(segment.getLabel() + ","); for (double feature : segment.getFeatures()) { output.append(feature + ","); } output.append(set + "\n"); } return output.toString(); }