HF RL Explorer

Task: Information Gain Calculation with Scikit-learn

Task: Information Gain Calculation with Scikit-learn: a task in Terminal-Lego-15k (Harbor dataset). Implement a Python script that calculates the Information Gain for each attribute (feature) with respect to a class label, using scikit-learn's mutual information functionality. Information Gain is…

The task

Implement a Python script that calculates the Information Gain for each attribute (feature) with respect to a class label, using scikit-learn's mutual information functionality. Information Gain is defined as `H(Class) - H(Class | Attribute)`, which is equivalent to mutual information between the attribute and the…

Part of PrimeIntellect/Terminal-Lego-15k.