6eaddaa9c3
* [CI] Fix CI with updated dependencies. - Fix jvm package get iris. * Skip SHAP test for now. * Revert "Skip SHAP test for now." This reverts commit 9aa28b4d8aee53fa95d92d2a879c6783ff4b2faa. * Catch all exceptions.
11 lines
434 B
Python
11 lines
434 B
Python
import numpy as np
|
|
import pandas
|
|
from sklearn.datasets import load_iris
|
|
|
|
X, y = load_iris(return_X_y=True)
|
|
y = y.astype(np.int32)
|
|
df = pandas.DataFrame(data=X, columns=['sepal length', 'sepal width', 'petal length', 'petal width'])
|
|
class_id_to_name = {0:'Iris-setosa', 1:'Iris-versicolor', 2:'Iris-virginica'}
|
|
df['class'] = np.vectorize(class_id_to_name.get)(y)
|
|
df.to_csv('./iris.csv', float_format='%.1f', header=False, index=False)
|