Example usage for weka.core Capabilities enableDependency

Introduction

In this page you can find the example usage for weka.core Capabilities enableDependency.

Prototype

public void enableDependency(Capability c) 

Source Link

Document

enables the dependency flag for the given capability Enabling NOMINAL_ATTRIBUTES also enables BINARY_ATTRIBUTES, UNARY_ATTRIBUTES and EMPTY_NOMINAL_ATTRIBUTES.

Usage

From source file:WrapperSubset.java

License:Open Source License

/**
 * Returns the capabilities of this evaluator.
 *
 * @return the capabilities of this evaluator
 * @see Capabilities/*from ww w . j av  a  2s  . c o m*/
 */
@Override
public Capabilities getCapabilities() {
    Capabilities result;

    if (getClassifier() == null) {
        result = super.getCapabilities();
        result.disableAll();
    } else {
        result = getClassifier().getCapabilities();
    }

    // set dependencies
    for (Capability cap : Capability.values()) {
        result.enableDependency(cap);
    }

    // adjustment for class based on selected evaluation metric
    result.disable(Capability.NUMERIC_CLASS);
    result.disable(Capability.DATE_CLASS);
    if (m_evaluationMeasure != EVAL_ACCURACY && m_evaluationMeasure != EVAL_FMEASURE
            && m_evaluationMeasure != EVAL_AUC && m_evaluationMeasure != EVAL_AUPRC) {
        result.enable(Capability.NUMERIC_CLASS);
        result.enable(Capability.DATE_CLASS);
    }

    result.setMinimumNumberInstances(getFolds());

    return result;
}

---

From source file:LogReg.FilteredLogRegClassifier.java

License:Open Source License

/**
 * Returns default capabilities of the classifier.
 *
 * @return      the capabilities of this classifier
 *//*from w w  w.j a  va2  s  . co  m*/
public Capabilities getCapabilities() {
    Capabilities result;

    if (getFilter() == null)
        result = super.getCapabilities();
    else
        result = getFilter().getCapabilities();

    // the filtered classifier always needs a class
    result.disable(Capability.NO_CLASS);

    // set dependencies
    for (Capability cap : Capability.values())
        result.enableDependency(cap);

    return result;
}

---

From source file:ml.engine.LibSVM.java

License:Open Source License

/**
 * Returns default capabilities of the classifier.
 * /*from   w ww .j a v a2 s.  c  o  m*/
 * @return the capabilities of this classifier
 */
@Override
public Capabilities getCapabilities() {
    Capabilities result = super.getCapabilities();
    result.disableAll();

    // attributes
    result.enable(Capability.NOMINAL_ATTRIBUTES);
    result.enable(Capability.NUMERIC_ATTRIBUTES);
    result.enable(Capability.DATE_ATTRIBUTES);
    result.enable(Capability.MISSING_VALUES);

    // class
    result.enableDependency(Capability.UNARY_CLASS);
    result.enableDependency(Capability.NOMINAL_CLASS);
    result.enableDependency(Capability.NUMERIC_CLASS);
    result.enableDependency(Capability.DATE_CLASS);

    switch (m_SVMType) {
    case SVMTYPE_C_SVC:
    case SVMTYPE_NU_SVC:
        result.enable(Capability.NOMINAL_CLASS);
        break;

    case SVMTYPE_ONE_CLASS_SVM:
        result.enable(Capability.UNARY_CLASS);
        break;

    case SVMTYPE_EPSILON_SVR:
    case SVMTYPE_NU_SVR:
        result.enable(Capability.NUMERIC_CLASS);
        result.enable(Capability.DATE_CLASS);
        break;

    default:
        throw new IllegalArgumentException("SVMType " + m_SVMType + " is not supported!");
    }
    result.enable(Capability.MISSING_CLASS_VALUES);

    return result;
}

---