Class LevelVectors

src/com/winvector/variables/LevelVectors.java:15–148

@author johnmount

 *
 */
public final class LevelVectors implements VariableMapping {
	private static final long serialVersionUID = 1L;
	
	private final String origColumn;
	private final int index;
	private final Map<String,VectorRow> levelCodes; // level to vector of codes
	private final int targetDim;

	public static final class VectorRow implements Serializable {
		private static final long serialVersionUID = 1L;
		public final String[] names;
		public final double[] levelEncodings;
		public final int[] warmStartOutcome;
		
		public VectorRow(final String[] name, final double[] levelEncodings, final int[] warmStartOutcome) {
			this.names = name;
			this.levelEncodings = levelEncodings;
			this.warmStartOutcome = warmStartOutcome;
		}
	}
	
	public LevelVectors(final String origColumn, final int index, 
			final Map<String,VectorRow> levelCodes) { 
		this.origColumn = origColumn;
		this.index = index;
		this.levelCodes = levelCodes;
		if(!levelCodes.isEmpty()) {
			targetDim = levelCodes.values().iterator().next().levelEncodings.length;
		} else {
			targetDim = 0;
		}
	}
	
	@Override
	public String origColumn() {
		return origColumn;
	}

	@Override
	public int indexL() {
		return index;
	}

	@Override
	public int indexR() {
		return index + targetDim;
	}
	
	@Override
	public boolean wantRegularization() {
		return false;
	}

	@Override
	public void process(final BurstMap row, final double[] vec) {
		final String level = row.getAsString(origColumn);
		if(level!=null) {
			final VectorRow code = levelCodes.get(level);