All Downloads are FREE. Search and download functionalities are using the official Maven repository.

ml.bundle.support.v1.json.MlJsonSerializerSupport.scala Maven / Gradle / Ivy

The newest version!
package ml.bundle.support.v1.json

import ml.bundle.StreamSerializer
import ml.bundle.v1.{core, runtime}
import ml.bundle.support.JsonStreamSerializer.ConversionOps
import MlJsonSupport._

/**
  * Created by hollinwilkins on 3/29/16.
  */
trait MlJsonSerializerSupport {
  val mlNodeMetaDataSerializer: StreamSerializer[core.tree.node.MetaData.MetaData] =
    mlNodeMetaDataFormat.toStreamSerializer("ml.bundle.v1.core.tree.node.MetaData")
  val mlDecisionTreeMetaDataSerializer: StreamSerializer[core.regression.decision_tree.MetaData.MetaData] =
    mlDecisionTreeMetaDataFormat.toStreamSerializer("ml.bundle.v1.core.regression.decision_tree.MetaData")
  val mlRandomForestMetaDataSerializer: StreamSerializer[core.regression.random_forest.MetaData.MetaData] =
    mlRandomForestMetaDataFormat.toStreamSerializer("ml.bundle.v1.core.regression.random_forest.MetaData")
  val mlRandomForestClassificationMetaDataSerializer: StreamSerializer[core.classification.random_forest.MetaData.MetaData] =
    mlRandomForestClassificationMetaDataFormat.toStreamSerializer("ml.bundle.v1.core.classification.random_forest.MetaData")
  val mlDecisionTreeClassificationMetaDataSerializer: StreamSerializer[core.classification.decision_tree.MetaData.MetaData] =
    mlDecisionTreeClassificationMetaDataFormat.toStreamSerializer("ml.bundle.v1.core.classification.decision_tree.MetaData")

  val mlNodeDataSerializer: StreamSerializer[core.tree.node.NodeData.NodeData] =
    mlNodeDataFormat.toStreamSerializer("ml.bundle.v1.core.tree.node.NodeData")
  val mlRandomForestRegressionModelMetaDataSerializer: StreamSerializer[runtime.regression.random_forest.MetaData.MetaData] =
    mlRandomForestRegressionModelMetaDataFormat.toStreamSerializer("ml.bundle.v1.runtime.regression.random_forest.MetaData")
  val mlLinearRegressionModelSerializer: StreamSerializer[runtime.regression.LinearRegressionModel.LinearRegressionModel] =
    mlLinearRegressionModelFormat.toStreamSerializer("ml.bundle.v1.runtime.regression.LinearRegressionModel")

  val mlLogisticRegressionModelSerializer: StreamSerializer[runtime.classification.LogisticRegressionModel.LogisticRegressionModel] =
    mlLogisticRegressionModelFormat.toStreamSerializer("ml.bundle.v1.runtime.classification.LogisticRegressionModel")
  val mlSupportVectorMachineModelSerializer: StreamSerializer[runtime.classification.SupportVectorMachineModel.SupportVectorMachineModel] =
    mlSupportVectorMachineModelFormat.toStreamSerializer("ml.bundle.v1.runtime.classification.SupportVectorMachineModel")
  val mlRandomForestClassificationModelMetaDataSerializer: StreamSerializer[runtime.classification.random_forest.MetaData.MetaData] =
    mlRandomForestClassificationModelMetaDataFormat.toStreamSerializer("ml.bundle.v1.runtime.classification.random_forest.MetaData")

  val mlOneHotEncoderModelSerializer: StreamSerializer[runtime.feature.OneHotEncoderModel.OneHotEncoderModel] =
    mlOneHotEncoderModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.OneHotEncoderModel")
  val mlHashingTermFrequencyModelSerializer: StreamSerializer[runtime.feature.HashingTermFrequencyModel.HashingTermFrequencyModel] =
    mlHashingTermFrequencyModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.HashingTermFrequencyModel")
  val mlStandardScalerModelSerializer: StreamSerializer[runtime.feature.StandardScalerModel.StandardScalerModel] =
    mlStandardScalerModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.StandardScalerModel")
  val mlStringIndexerModelSerializer: StreamSerializer[runtime.feature.StringIndexerModel.StringIndexerModel] =
    mlStringIndexerModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.StringIndexerModel")
  val mlReverseStringIndexerModelSerializer: StreamSerializer[runtime.feature.ReverseStringIndexerModel.ReverseStringIndexerModel] =
    mlReverseStringIndexerModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.ReverseStringIndexerModel")
  val mlTokenizerModelSerializer: StreamSerializer[runtime.feature.TokenizerModel.TokenizerModel] =
    mlTokenizerModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.TokenizerModel")
  val mlVectorAssemblerModelSerializer: StreamSerializer[runtime.feature.VectorAssemblerModel.VectorAssemblerModel] =
    mlVectorAssemblerModelFormat.toStreamSerializer("ml.bundle.v1.runtime.feature.VectorAssemblerModel")

  val mlPipelineModelMetaDataSerializer: StreamSerializer[runtime.pipeline_model.MetaData.MetaData] =
    mlPipelineModelMetaDataFormat.toStreamSerializer("ml.bundle.v1.runtime.pipeline_model.MetaData")
}
object MlJsonSerializerSupport extends MlJsonSerializerSupport




© 2015 - 2024 Weber Informatics LLC | Privacy Policy