Learned embeddings from deep learning to visualize and predict protein sets