I have a dataset for people doing trade in various segments (classes) .I am trying to build a multi-label classifier to predict people trading in various segments (classes).

My dataset :

Client_id [Demographic data] orders_perday Traded_days Segment
 A123      ..............     3             4           equity
 A123      ..............     2             2           commodity
 A123      ..............     1             9           currency
 B789      ..............     7             8           equity
 B789      ..............     3             2           futures
 C456      ..............     2             7           currency
 D987      ..............    10             1           equity
 C183      ..............     2             9           currency

Demographic data includes age, sex, city, income ,etc.

I need to build a model which will predict new clients belong to which class(segment) can be 1 or more. Please give some advice of how to approach this problem


As with any data science problem: Approach the problem by first exploring the data and then trying the simplest solution that you think might just work, to some extent.

To explore the data, look at the distributions of all the features. Are they at all helpful to identifying the segments? How many trades do you have in the segments, are the classes balanced (more or less similar counts in all classes)? If not, you need to be careful with a metric like accuracy and you may be better off using F1 score, for example.

For an algorithm, how about starting with a decision tree and taking it from there? You can use it as a baseline model to compare to the results of other experiments. For example more complex algorithms using trees like boosted trees or random forest, or neural networks.

The performance metric is often averaged across all classes, using a technique called “1 versus all”. For that you start with the first segment, like “equity”. You can use the model to identify “equity” or “not equity”, and calculate the F1 score for this task. Repeat for all other classes and average.


Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.