<jupyter_start><jupyter_text>Split positive and negative class<jupyter_code>by_class = df_train.groupby('Best Performance')
df_train_positive = by_class.get_group(1)
df_train_negative = by_class.get_group(0)
print(df_train_positive.shape)
print(df_train_negative.shape)
df_train_positive.to_csv("1.train_positive.csv", index=False)
df_train_negative.to_csv("1.train_negative.csv", index=False)<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.11347517730496454, 'fraction_numerical': 0.009456264775413711, 'mean_word_length': 12.67741935483871, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 5, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '1', 'dir': 'jupyter-structured-clean-dedup', 'id': '597522', 'license': 'MIT', 'n_tokens_mistral': 161, 'n_tokens_neox': 155, 'n_words': 17, 'path': '1. Splitting based on class.ipynb', 'repo_name': 'vhiefa/People-Analytics-BRI-Challenge-EDA', 'size': '423'}