com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict Class Reference

Inheritance diagram for com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict:

Collaboration diagram for com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict:

Public Member Functions
	MLPredict ()
RelDataType	inferReturnType (SqlOperatorBinding opBinding)
SqlOperandCountRange	getOperandCountRange ()
boolean	checkOperandTypes (SqlCallBinding callBinding, boolean throwOnFailure)

Detailed Description

Definition at line 747 of file HeavyDBSqlOperatorTable.java.

Constructor & Destructor Documentation

com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict.MLPredict ( )

inline

Definition at line 748 of file HeavyDBSqlOperatorTable.java.

                       {
      super("ML_PREDICT",
              SqlKind.OTHER_FUNCTION,
              null,
              null,
              null,
              SqlFunctionCategory.SYSTEM);
    }

Member Function Documentation

boolean com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict.checkOperandTypes ( SqlCallBinding  callBinding, boolean  throwOnFailure  )

inline

Definition at line 770 of file HeavyDBSqlOperatorTable.java.

                                                                                         {
      // A call to ML_PREDICT can take the following arguemnts
      // 1. A text literal with the model name - REQUIRED
      // 2. Any number of optional text column arguments
      // 3. Any number of numeric arguments
      final SqlValidator validator = callBinding.getValidator();

      final int num_operands = callBinding.getOperandCount();
      if (num_operands < 2) {
        throw new IllegalArgumentException(
                "At least 2 arguments are required, the model name and one or more predictors.");
      }
      for (int operand_idx = 0; operand_idx < num_operands; operand_idx++) {
        final SqlNode operand = callBinding.operand(operand_idx);
        final SqlTypeName operand_type =
                validator.getValidatedNodeType(operand).getSqlTypeName();
        final SqlTypeFamily operand_type_family = operand_type.getFamily();
        if (operand_idx == 0) {
          if (!operand.isA(EnumSet.of(SqlKind.LITERAL))
                  || operand_type_family != SqlTypeFamily.CHARACTER) {
            throw new IllegalArgumentException(
                    "First argument must be TEXT literal denoting the model name.");
          }
        } else {
          if (operand.isA(EnumSet.of(SqlKind.LITERAL))) {
            throw new IllegalArgumentException(
                    "Literals are not supported as predictors.");
          }
          if (!(operand_type_family == SqlTypeFamily.NUMERIC
                      || operand_type_family == SqlTypeFamily.CHARACTER)) {
            throw new IllegalArgumentException(
                    "Only TEXT and NUMERIC predictors are supported.");
          }
        }
      }
      return true;
    }

SqlOperandCountRange com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict.getOperandCountRange ( )

inline

Definition at line 765 of file HeavyDBSqlOperatorTable.java.

                                                       {
      return SqlOperandCountRanges.from(2);
    }

RelDataType com.mapd.calcite.parser.HeavyDBSqlOperatorTable.MLPredict.inferReturnType ( SqlOperatorBinding  opBinding )

inline

Definition at line 758 of file HeavyDBSqlOperatorTable.java.

                                                                     {
      final RelDataTypeFactory typeFactory = opBinding.getTypeFactory();
      return typeFactory.createTypeWithNullability(
              typeFactory.createSqlType(SqlTypeName.DOUBLE), /*nullable=*/true);
    }

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/java/calcite/src/main/java/com/mapd/calcite/parser/HeavyDBSqlOperatorTable.java