# Regression instead of a classification

**URL:** <https://community.konduit.ai/t/regression-instead-of-a-classification/595>\
**Category:** DL4J\
**Created:** [June 8, 2020, 8:11am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595 "2020-06-08T08:11:17Z")\
**Posts on this page:** 7\
**Page:** 1

<div class="post-metadata">

**Author:** ![Jolosin4](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/jolosin4/32/223_2.png) [@Jolosin4](https://community.konduit.ai/u/Jolosin4)\
**Post date:** [June 8, 2020, 8:11am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/1 "2020-06-08T08:11:17Z")

</div>

I have a question. How can I write a regression instead of a categorical classification like:  
.classification(finalSchema.getIndexOfColumn(“STOCK”), 2)  
The Column STOCK has an input like this: 78941079, 78941015, 78941011, 78941025…?

---

<div class="post-metadata">

**Author:** ![treo](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/treo/32/47_2.png) [@treo](https://community.konduit.ai/u/treo)\
**Post date:** [June 8, 2020, 8:21am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/2 "2020-06-08T08:21:07Z")

</div>

You tell it to use regression instead of classification:  
[https://deeplearning4j.org/api/latest/org/deeplearning4j/datasets/datavec/RecordReaderDataSetIterator.Builder.html](https://deeplearning4j.org/api/latest/org/deeplearning4j/datasets/datavec/RecordReaderDataSetIterator.Builder.html)

And then you set up your model to use an appropriate activation function (most likely Identity) and loss function (most likely MSE).

---

<div class="post-metadata">

**Author:** ![Jolosin4](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/jolosin4/32/223_2.png) [@Jolosin4](https://community.konduit.ai/u/Jolosin4)\
**Post date:** [June 8, 2020, 8:49am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/3 "2020-06-08T08:49:17Z")

</div>

labelIndexFrom = 0, labelIndexTo = 3  
RecordReaderDataSetIterator testIterator = new RecordReaderDataSetIterator.Builder(testRecordReader, batchSize)  
.regression(0, 3)  
.build

???

---

<div class="post-metadata">

**Author:** ![treo](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/treo/32/47_2.png) [@treo](https://community.konduit.ai/u/treo)\
**Post date:** [June 8, 2020, 8:51am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/4 "2020-06-08T08:51:28Z")

</div>

What is your question here? I’m not a compiler.

---

<div class="post-metadata">

**Author:** ![Jolosin4](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/jolosin4/32/223_2.png) [@Jolosin4](https://community.konduit.ai/u/Jolosin4)\
**Post date:** [June 8, 2020, 8:53am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/5 "2020-06-08T08:53:11Z")

</div>

sorry I dont know what you mean with the appropriate activation function and the loss function? Could you give me any example?

---

<div class="post-metadata">

**Author:** ![treo](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/treo/32/47_2.png) [@treo](https://community.konduit.ai/u/treo)\
**Post date:** [June 8, 2020, 9:09am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/6 "2020-06-08T09:09:50Z")

</div>

There are a few examples on how to set up the model for regression in the examples repository:  
[https://github.com/eclipse/deeplearning4j-examples/tree/master/dl4j-examples/src/main/java/org/deeplearning4j/examples/feedforward/regression](https://github.com/eclipse/deeplearning4j-examples/tree/master/dl4j-examples/src/main/java/org/deeplearning4j/examples/feedforward/regression)

Given how you’ve stated your original question, I understood it to be about loading the data for regression. This is a separate step from setting up your model for it.

---

<div class="post-metadata">

**Author:** ![Jolosin4](https://yyz1.discourse-cdn.com/flex035/user_avatar/community.konduit.ai/jolosin4/32/223_2.png) [@Jolosin4](https://community.konduit.ai/u/Jolosin4)\
**Post date:** [June 8, 2020, 9:14am UTC](https://community.konduit.ai/t/regression-instead-of-a-classification/595/7 "2020-06-08T09:14:26Z")

</div>

ok thank you very much!