Files
aiRtrafficNN/Code/python/notebooks/02_test_rnn.ipynb
T
2025-11-20 12:36:30 +01:00

98 lines
1.9 KiB
Plaintext

{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"id": "276487d8",
"metadata": {},
"outputs": [],
"source": [
"import torch\n",
"from aiRNN import dataloader, models, losses\n",
"import pathlib"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "1308b1d5",
"metadata": {},
"outputs": [],
"source": [
"file_list = pathlib.Path(\"/home/lars/Documents/Studium/UiO/data_analysis/project3/Code/cpp/known_routes_and_aircraft.csv\")\n",
"base_path = file_list.parent\n",
"file_list = file_list.read_text().splitlines()\n",
"file_list = [(base_path / f).resolve() for f in file_list]"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "0d8e6107",
"metadata": {},
"outputs": [],
"source": [
"dataset = dataloader.EvenlySpacedDataset(\n",
" filepaths=file_list[:50],\n",
" n_input=30*10, # 10 minutes input\n",
" n_output=30*5, # 5 minutes output\n",
" n_windows_per_file=5,\n",
" step=1,\n",
")"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "595103c9",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"235"
]
},
"execution_count": 4,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"len(dataset)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "522cd04f",
"metadata": {},
"outputs": [],
"source": [
"test_model = models."
]
}
],
"metadata": {
"kernelspec": {
"display_name": "adsbpy",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.13.9"
}
},
"nbformat": 4,
"nbformat_minor": 5
}