recodehive / Stackoverflow-Analysis

Stack overflow is a professional community for developers. This repo analysis 3 years of developer Survey done by Stackoverflow and do visualization and predict the salary of Data Scientist in future.
https://stackoverflow-analysis.streamlit.app/
MIT License
220 stars 108 forks source link

[Feature Request]: Add KNN regression #270

Closed Aditi22Bansal closed 3 months ago

Aditi22Bansal commented 3 months ago

Is there an existing issue for this?

Feature Description

In this, first standardize the features using StandardScaler to ensure that all features have the same scale. Then, train a KNN regression model with 5 neighbors (n_neighbors=5). Then predict the test set results and evaluate the model using mean squared error and R^2 score

Use Case

Integrating a K-Nearest Neighbors (KNN) regression model alongside the existing linear regression model adds depth and versatility to the analysis of salary prediction in the developer community dataset.

Benefits

By incorporating a KNN regression model alongside the existing linear regression model, the project can potentially achieve higher predictive accuracy. KNN regression is particularly useful for capturing non-linear relationships in the data, which may lead to more accurate salary predictions. The addition of KNN regression enables a more comprehensive exploration of the relationships between features and salaries in the developer community dataset. By leveraging the local information from neighboring data points, KNN regression can uncover subtle patterns and nuances that may be overlooked by linear regression.

Priority

High

Record

github-actions[bot] commented 3 months ago

Hi there! Thanks for opening this issue. We appreciate your contribution to this open-source project. We aim to respond or assign your issue as soon as possible.

Aditi22Bansal commented 3 months ago

@sanjay-kv Kindly assign me this issue

github-actions[bot] commented 3 months ago

Hello @Aditi22Bansal! Your issue #270 has been closed. Thank you for your contribution!