search

dmmiller612 / sparktorch

Train and run Pytorch models on Apache Spark.
MIT License
339 stars 44 forks source link

getting vectors #19

Open lizush23 opened 4 years ago

lizush23 commented 4 years ago

Hi I would like to convert pytorch pretrained model into spark and use it feature extraction. is this the right syntax

spark_torch_model = create_spark_torch_model( net, inputCol='Message',

predictionCol='predictions',

useVectorOut=True

)

and can i skip predictions as all i need is vectors? thanks