• Home
• com.flipkart.fdp.ml
• models-info
• 0.1.2
• source code
• LogisticRegressionModelInfo.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.modelinfo.LogisticRegressionModelInfo Maven / Gradle / Ivy

package com.flipkart.fdp.ml.modelinfo;

import com.flipkart.fdp.ml.transformer.LogisticRegressionTransformer;
import com.flipkart.fdp.ml.transformer.Transformer;
import lombok.Data;

/**
 * Represents information for a Logistic Regression model
 */
@Data
public class LogisticRegressionModelInfo extends AbstractModelInfo {
    private double[] weights;
    private double intercept;
    private int numClasses;
    private int numFeatures;
    private double threshold;

    private String probabilityKey = "probability";

    /**
     * @return an corresponding {@link LogisticRegressionTransformer} for this model info
     */
    @Override
    public Transformer getTransformer() {
        return new LogisticRegressionTransformer(this);
    }
}