Data Science, Artificial Intelligence, AI, Machine Learning, Python.
dataset['train_input'] = torch.from_numpy(train_input) dataset['test_input'] = torch.from_numpy(test_input) dataset['train_label'] = torch.from_numpy(train_label) dataset['test_label'] = torch.from_numpy(test_label) X = dataset['train_input'] y = dataset['train_label'] plt.scatter(X[:,0], X[:,1],