org.deeplearning4j.ui.components.chart.ChartScatter Maven / Gradle / Ivy
/*******************************************************************************
* Copyright (c) 2015-2018 Skymind, Inc.
*
* This program and the accompanying materials are made available under the
* terms of the Apache License, Version 2.0 which is available at
* https://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*
* SPDX-License-Identifier: Apache-2.0
******************************************************************************/
package org.deeplearning4j.ui.components.chart;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.deeplearning4j.ui.components.chart.style.StyleChart;
import org.nd4j.shade.jackson.annotation.JsonInclude;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**Scatter chart
*
* @author Alex Black
*/
@EqualsAndHashCode(callSuper = true)
@Data
@JsonInclude(JsonInclude.Include.NON_NULL)
public class ChartScatter extends Chart {
public static final String COMPONENT_TYPE = "ChartScatter";
private List x;
private List y;
private List seriesNames;
private ChartScatter(Builder builder) {
super(COMPONENT_TYPE, builder);
x = builder.x;
y = builder.y;
seriesNames = builder.seriesNames;
}
public ChartScatter() {
super(COMPONENT_TYPE);
//no-arg constructor for Jackson
}
public static class Builder extends Chart.Builder {
private List x = new ArrayList<>();
private List y = new ArrayList<>();
private List seriesNames = new ArrayList<>();
public Builder(String title, StyleChart style) {
super(title, style);
}
/**
*
* @param seriesName Name of the series
* @param xValues Array of x values
* @param yValues Array of y values (such that a single point i has coordinates (x[i],y[i]))
* @return
*/
public Builder addSeries(String seriesName, double[] xValues, double[] yValues) {
x.add(xValues);
y.add(yValues);
seriesNames.add(seriesName);
return this;
}
public ChartScatter build() {
return new ChartScatter(this);
}
}
@Override
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append("ChartScatter(x=[");
boolean first = true;
if (x != null) {
for (double[] d : x) {
if (!first)
sb.append(",");
sb.append(Arrays.toString(d));
first = false;
}
}
sb.append("],y=[");
first = true;
if (y != null) {
for (double[] d : y) {
if (!first)
sb.append(",");
sb.append(Arrays.toString(d));
first = false;
}
}
sb.append("],seriesNames=");
if (seriesNames != null)
sb.append(seriesNames);
sb.append(")");
return sb.toString();
}
}