• Home
• tech.mlsql
• streamingpro-mlsql-spark_2.4
• 1.2.0
• source code
• sk_predict.py

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

python.sk_predict.py Maven / Gradle / Ivy

from pyspark.ml.linalg import VectorUDT, Vectors
import pickle
import python_fun


def predict(index, s):
    items = [i for i in s]
    feature = VectorUDT().deserialize(pickle.loads(items[0]))
    model = pickle.loads(pickle.loads(items[1])[0])
    y = model.predict([feature.toArray()])
    return [VectorUDT().serialize(Vectors.dense(y))]


python_fun.udf(predict)