Decision trees notebook line 10 has assignment issue for splitting the dataset

[ashna_upadhyay]

Hi,

I was going through the Decision trees notebook exercise 2 under AL_Kit_XGBoost and it looks like there is some assignment issue. Please refer to the screenshot attached.

I believe there should be 2 modifications

1. Current code: strat_shuff_split = xxx(n_splits=1, test_size=1000, random_state=42)
    Should be modified to: xxx = StratifiedShuffleSplit(n_splits=1, test_size=1000, random_state=42)

2. Current code: train_idx, test_idx = next(strat_shuff_split.split(data[feature_cols], data['color']))
     Should be modified to: train_idx, test_idx = next(xxx.split(data[feature_cols], data['color']))

Please let me know if you need any further information.

Thanks,

Ashna

[AthiraM_Intel]

Hi,

Thank you for posting in Intel Communities.

Could you please let us know which sample you are trying?

From where you are getting the samples? Are these samples a part of any Intel workshop?

Thanks.

[ashna_upadhyay]

Hi,

I am trying the samples under Intel's training modules, please refer to the screenshot below.

The notebook is under AL_Kit_XGBoost_Predictive_Modeling > Decision Trees.

Thanks!

[AthiraM_Intel · Accepted solution]

Hi,

The notebook that you are referring to is an exercise notebook. You are expected to understand the concepts and fill out the xxx with the required module, function, or variable names.

If this resolves your issue, make sure to accept this as a solution. This would help others with similar issues.

Thanks

[AthiraM_Intel]

Hi,

Thanks for accepting our solution. If you need any additional information, please post a new question as this thread will no longer be monitored by Intel.

Thanks