pytorch-frame icon indicating copy to clipboard operation
pytorch-frame copied to clipboard

Add ColumnWiseEncoder

Open yiweny opened this issue 2 years ago • 1 comments

So we can assign a different encoder for different columns.

yiweny avatar Nov 20 '23 03:11 yiweny

Thanks for raising this! Let me take on this. I think a promising approach is to add an argument, col_encoder_dict, in StypeWiseEncoder so that it uses the given StypeEncoder for specific columns while using the default StypeEncoder (specified in stype_encoder_dict) for the rest.

We can support ColumnWiseEncoder, but it sounds a bit overkill to specify the encoder for each column name.

weihua916 avatar Nov 21 '23 08:11 weihua916