Machine Learning, Sentiment Analysis, Vertex AI, Programming, Bert.
stratify=df['Target'], test_size=0.2) df_eval, df_test = train_test_split(df_, stratify=df_['Target'], test_size=0.5) print(f"Files will be saved in {output_dir}") df_train.to_csv(output_dir + "/train.csv", index=False) df_eval.to_csv(output_dir + "/eval.csv", index=False) df_test.to_csv(output_dir