net.sansa_stack.ml.spark.utils.NodeIndexer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sansa-ml-spark_2.12 Show documentation

RDF/OWL Machine Learning Library for Apache Spark

The newest version!

package net.sansa_stack.ml.spark.utils

import org.apache.jena.graph.Node
import org.apache.spark.sql.DataFrame

abstract class NodeIndexer {
  protected var vocabulary: Map[Node, Int]

  val inputCols: Array[String] = Array("s", "p", "o")



  def fit(df: DataFrame): NodeIndexerModel = {

    // val allNodes = df.select(inputCols)
    // println(allNodes)

    new NodeIndexerModel
  }
}

class NodeIndexerModel {
  def transform(df: DataFrame): DataFrame = {
    df
  }
}