Skip to content

Commit 0e95b60

Browse files
author
Vladimir Iglovikov
committed
Added method points_to_csv that saves known data points to csv file
1 parent eda6bed commit 0e95b60

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

bayes_opt/bayesian_optimization.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,7 @@ def points_to_csv(self, file_name):
337337
338338
:return: None
339339
"""
340-
import pandas as pd
341-
points_df = pd.DataFrame(self.X, columns=self.keys)
342-
points_df['target'] = self.Y
343-
points_df.to_csv(file_name, index=False)
340+
341+
points = np.hstack((self.X, np.expand_dims(self.Y, axis=1)))
342+
header = ', '.join(self.keys + ['target'])
343+
np.savetxt(file_name, points, header=header, delimiter=',')

0 commit comments

Comments
 (0)