Skip to content

Commit dec0910

Browse files
committed
Migrate column accessors usage in titanic example
1 parent 4a59caa commit dec0910

File tree

1 file changed

+2
-3
lines changed
  • examples/idea-examples/titanic/src/main/kotlin/org/jetbrains/kotlinx/dataframe/examples/titanic/ml

1 file changed

+2
-3
lines changed

examples/idea-examples/titanic/src/main/kotlin/org/jetbrains/kotlinx/dataframe/examples/titanic/ml/titanic.kt

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,13 +79,12 @@ private fun <T> OnHeapDataset.Companion.create(
7979
dataframe: DataFrame<T>,
8080
yColumn: ColumnSelector<T, Number>,
8181
): OnHeapDataset {
82-
val x by column<FloatArray>("X")
8382

8483
fun extractX(): Array<FloatArray> =
8584
dataframe.remove(yColumn)
8685
.convert { colsAtAnyDepth { !it.isColumnGroup() } }.toFloat()
87-
.merge { colsAtAnyDepth().colsOf<Float>() }.by { it.toFloatArray() }.into(x)
88-
.getColumn(x).toTypedArray()
86+
.merge { colsAtAnyDepth().colsOf<Float>() }.by { it.toFloatArray() }.into("X")
87+
.getColumn("X").cast<FloatArray>().toTypedArray()
8988

9089
fun extractY(): FloatArray = dataframe.get(yColumn).toFloatArray()
9190

0 commit comments

Comments
 (0)