forked from D-Net/dnet-hadoop
67 lines
1.3 KiB
Java
67 lines
1.3 KiB
Java
package eu.dnetlib.pace.model;
|
|
|
|
import java.io.Serializable;
|
|
import java.lang.reflect.InvocationTargetException;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import com.google.gson.Gson;
|
|
import eu.dnetlib.pace.clustering.*;
|
|
|
|
public class ClusteringDef implements Serializable {
|
|
|
|
private String name;
|
|
|
|
private List<String> fields;
|
|
|
|
private Map<String, Integer> params;
|
|
|
|
private ClusteringResolver clusteringResolver = new ClusteringResolver();
|
|
|
|
public ClusteringDef() {}
|
|
|
|
public String getName() {
|
|
return name;
|
|
}
|
|
|
|
public void setName(final String name) {
|
|
this.name = name;
|
|
}
|
|
|
|
public ClusteringFunction getClusteringFunction() {
|
|
|
|
try {
|
|
ClusteringFunction clusteringFunction = clusteringResolver.resolve(getName());
|
|
clusteringFunction.setParams(params);
|
|
return clusteringFunction;
|
|
|
|
} catch (IllegalAccessException | InstantiationException | NoSuchMethodException | InvocationTargetException e) {
|
|
e.printStackTrace();
|
|
return new RandomClusteringFunction(getParams());
|
|
}
|
|
|
|
}
|
|
|
|
public List<String> getFields() {
|
|
return fields;
|
|
}
|
|
|
|
public void setFields(final List<String> fields) {
|
|
this.fields = fields;
|
|
}
|
|
|
|
public Map<String, Integer> getParams() {
|
|
return params;
|
|
}
|
|
|
|
public void setParams(final Map<String, Integer> params) {
|
|
this.params = params;
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
return new Gson().toJson(this);
|
|
}
|
|
|
|
}
|