com.flipkart.fdp.ml.adapter.MinMaxScalerModelInfoAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of adapters-1.6_2.10 Show documentation

The newest version!

package com.flipkart.fdp.ml.adapter;

import com.flipkart.fdp.ml.modelinfo.MinMaxScalerModelInfo;
import org.apache.spark.ml.feature.MinMaxScalerModel;
import org.apache.spark.sql.DataFrame;

import java.util.LinkedHashSet;
import java.util.Set;

/**
 * Transforms Spark's {@link MinMaxScalerModel} in MlLib to  {@link com.flipkart.fdp.ml.modelinfo.MinMaxScalerModelInfo} object
 * that can be exported through {@link com.flipkart.fdp.ml.export.ModelExporter}
 */
public class MinMaxScalerModelInfoAdapter extends AbstractModelInfoAdapter {
    @Override
    public MinMaxScalerModelInfo getModelInfo(final MinMaxScalerModel from, final DataFrame df) {
        final MinMaxScalerModelInfo modelInfo = new MinMaxScalerModelInfo();
        modelInfo.setOriginalMax(from.originalMax().toArray());
        modelInfo.setOriginalMin(from.originalMin().toArray());
        modelInfo.setMax(from.getMax());
        modelInfo.setMin(from.getMin());

        Set inputKeys = new LinkedHashSet();
        inputKeys.add(from.getInputCol());
        modelInfo.setInputKeys(inputKeys);

        Set outputKeys = new LinkedHashSet();
        outputKeys.add(from.getOutputCol());
        modelInfo.setOutputKeys(outputKeys);

        return modelInfo;
    }

    @Override
    public Class getSource() {
        return MinMaxScalerModel.class;
    }

    @Override
    public Class getTarget() {
        return MinMaxScalerModelInfo.class;
    }
}