# How to wisely choose machine algorithm for the given data set?

I have a data sample of RSSI value from three different devices and based on the RSSI sample, it should tell from which location the data sample is arrived. Below are the sample data set,

 device_1  device_2  device_3  location
-45       -56       -78       drawing_room
-48       -51       -82       drawing_room
-41       -59       -73       drawing_room
-71       -59       -59       conference
-69       -60       -65       conference
-73       -60       -52       conference
-33       -68       -64       kitchen
-32       -66       -63       kitchen
-37       -67       -61       kitchen
-63       -48       -48       lab
-62       -48       -46       lab
-59       -48       -54       lab


I will have "n" No.of data samples for "m" No.of location. The actual data set can be find here.

I want to predict "location" from the d1 d2 d3. Based on what parameters (i.e., correlation matrix or visualization), the machine learning algorithm can be chosen for the given data set?

The visualization chart for the considered data set is From the visualization chart, Does it mean only the column "device_1" and "device_2" has a good separation among the location compared to other columns?

Note: If required, the data samples can be considered as positive.

• Could you state more clearly exactly what you want to predict, and what features you want to use to predict it? I.e. do you want to predict location from d1 d2 d3? Aug 28, 2019 at 6:52
• Yes, I want to predict location from d1 d2 d3 features Aug 28, 2019 at 7:04
• You can train a neural network to classify the location based on three numeric features d1, d2 and d3. Also, we can normalize the features. Aug 28, 2019 at 7:32
• By looking at the features I would guess that a simpler method might be sufficient. The plot of device1-device3 is almost linearly seperable, thus linear seperators should get a reasonable level of correct predictions. You stated in another comment that you tried SVM and Random Forest, so maybe you need to tune the hyperparameters there. Aug 28, 2019 at 10:09
• @Eulenfuchswiesel, Thank you for your comment. Let me try with the hyperparameters. Aug 28, 2019 at 10:29

I used a simple Neural Network with 3 Inputs - 1 hidden layer with 8 neurons - 3 Outputs (Labels) with k-fold crossvalidation in keras for your data set.

import pandas as pd
from keras.models import Sequential
from keras.layers import Dense
from keras.wrappers.scikit_learn import KerasClassifier
from keras.utils import np_utils
from sklearn.model_selection import cross_val_score
from sklearn.model_selection import KFold
from sklearn.preprocessing import LabelEncoder, scale
from sklearn.pipeline import Pipeline

path = r"User\train.csv"

data.location.value_counts()
dataset = data.values

x = dataset[:,0:3]
x_scaled = scale(x)
y = dataset[:,3]

encoder = LabelEncoder()
encoder.fit(y)
encoded_Y = encoder.transform(y)

dummy_y = np_utils.to_categorical(encoded_Y)

def model():
model = Sequential()

return model

estimator = KerasClassifier(build_fn=model, epochs=200, batch_size=5, verbose=0)
kfold = KFold(n_splits=10, shuffle=True)

results = cross_val_score(estimator, x_scaled, dummy_y, cv=kfold)
print("Model Prediction: %.2f%% (%.2f%%)" % (results.mean()*100, results.std()*100))


The used Model can categorize around 94% of the data correctly. Maybe this helps for the start.

Out:
Prediction: 93.95% (3.13%)


In data science and statistics in general you have to try and train several models to see which suits your dataset best (model benchmarking). In the end it is often a trade-off between computational costs and accuracy.

• Thank you for your suggestion. Let me evaluate this with the real time data set for prediction and check how the performance is. I have tried with SVM & Random Forest classification algorithm, 40 to 50% of the test samples are predicting the adjacent location. Thats why I thought to get some suggestion here. Aug 28, 2019 at 9:09