Open lizush23 opened 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',
useVectorOut=True
)
and can i skip predictions as all i need is vectors? thanks
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',
)
and can i skip predictions as all i need is vectors? thanks