本文整理汇总了Scala中org.apache.spark.sql.types.NumericType类的典型用法代码示例。如果您正苦于以下问题:Scala NumericType类的具体用法?Scala NumericType怎么用?Scala NumericType使用的例子?那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了NumericType类的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Scala代码示例。
示例1: ProtoConversions
//设置package包名称以及导入依赖的类
package org.tensorframes.dsl
import org.tensorflow.framework.{AttrValue, DataType, NodeDef}
import org.tensorframes.impl.SupportedOperations
import org.apache.spark.sql.types.NumericType
private[tensorframes] object ProtoConversions {
def getDType(nodeDef: NodeDef): DataType = {
val opt = Option(nodeDef.getAttr.get("T")).orElse(Option(nodeDef.getAttr.get("dtype")))
val v = opt.getOrElse(throw new Exception(s"Neither 'T' no 'dtype' was found in $nodeDef"))
v.getType
}
def getDType(sqlType: NumericType): DataType = {
SupportedOperations.opsFor(sqlType).tfType
}
def sqlTypeToAttrValue(sqlType: NumericType): AttrValue = {
AttrValue.newBuilder().setType(getDType(sqlType)).build()
}
def dataTypeToAttrValue(dataType: DataType): AttrValue = {
AttrValue.newBuilder().setType(dataType).build()
}
}