in scikit_learn_script_mode_local_training_and_serving/scikit_learn_script_mode_local_training_and_serving.py [0:0]
def download_training_and_eval_data():
if os.path.isfile('./data/iris.csv'):
print('Training and dataset exist. Skipping Download')
else:
print('Downloading training dataset')
# Load Iris dataset, then join labels and features
iris = datasets.load_iris()
joined_iris = np.insert(iris.data, 0, iris.target, axis=1)
# Create directory and write csv
os.makedirs("./data", exist_ok=True)
np.savetxt("./data/iris.csv", joined_iris, delimiter=",", fmt="%1.1f, %1.3f, %1.3f, %1.3f, %1.3f")
print('Downloading completed')