• Home
• com.flipkart.fdp.ml
• adapters-1.6_2.10
• 0.4.0
• source code
• FillNAValuesTransformerModelInfoAdapter.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

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

package com.flipkart.fdp.ml.adapter;

import com.flipkart.fdp.ml.FillNAValuesTransformer;
import com.flipkart.fdp.ml.modelinfo.FillNAValuesTransformerModelInfo;
import org.apache.spark.sql.DataFrame;

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

/**
 * Transforms {@link FillNAValuesTransformer} to  {@link FillNAValuesTransformerModelInfo} object
 * that can be exported through {@link com.flipkart.fdp.ml.export.ModelExporter}
 */
public class FillNAValuesTransformerModelInfoAdapter extends AbstractModelInfoAdapter {

    @Override
    public FillNAValuesTransformerModelInfo getModelInfo(final FillNAValuesTransformer from, DataFrame df) {

        final FillNAValuesTransformerModelInfo modelInfo = new FillNAValuesTransformerModelInfo();
        modelInfo.setNaValuesMap(from.getNAValueMap());

        Set inputKeys = new LinkedHashSet();
        inputKeys.addAll(from.getNAValueMap().keySet());
        modelInfo.setInputKeys(inputKeys);

        Set outputKeys = new LinkedHashSet();
        outputKeys.addAll(from.getNAValueMap().keySet());
        modelInfo.setOutputKeys(outputKeys);

        return modelInfo;
    }

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

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